Spaces:
Build error
Build 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
|