Pie31415 commited on
Commit
d57512d
·
1 Parent(s): b8df5f7

updated app

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -12,6 +12,9 @@ subprocess.check_call(["pip", "install", "torch==1.6.0+cu101",
12
  "torchvision==0.7.0+cu101",
13
  "-f", "https://download.pytorch.org/whl/torch_stable.html"])
14
 
 
 
 
15
  sys.path.append("./rome/")
16
  sys.path.append('./DECA')
17
 
 
12
  "torchvision==0.7.0+cu101",
13
  "-f", "https://download.pytorch.org/whl/torch_stable.html"])
14
 
15
+ print(torch.__version__)
16
+ print(torch.version.cuda)
17
+
18
  sys.path.append("./rome/")
19
  sys.path.append('./DECA')
20