Spaces:
Runtime error
Runtime error
fixed CUDA packages
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
@@ -6,6 +6,6 @@ kornia==0.4.0
|
|
6 |
numpy
|
7 |
matplotlib
|
8 |
pillow
|
9 |
-
torch
|
10 |
-
torchvision
|
11 |
easydict
|
|
|
6 |
numpy
|
7 |
matplotlib
|
8 |
pillow
|
9 |
+
https://download.pytorch.org/whl/cu101/torch-1.6.0%2Bcu101-cp38-cp38-linux_x86_64.whl
|
10 |
+
https://download.pytorch.org/whl/cu101/torchvision-0.7.0%2Bcu101-cp38-cp38-linux_x86_64.whl
|
11 |
easydict
|