Spaces:
Build error
Build error
updated app
Browse files
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 |
|