Create env.py

env.py

@@ -0,0 +1,10 @@
+import spaces
+import os
+import torch
+
+@spaces.GPU(duration=5)
+def check():
+    print(os.system("nvidia-smi"))
+    print(torch.version.cuda)
+
+check()