PUM4CH3N commited on
Commit
177a386
·
1 Parent(s): ceb457f

Add Env Test

Browse files
Files changed (1) hide show
  1. env.py +10 -0
env.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ import spaces
2
+ import os
3
+ import torch
4
+
5
+ @spaces.GPU(duration=5)
6
+ def check():
7
+ print(os.system("nvidia-smi"))
8
+ print(torch.version.cuda)
9
+
10
+ check()