Update requirements.txt
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
|
@@ -6,5 +6,5 @@ scipy==1.11.2
|
|
| 6 |
timm==0.5.4
|
| 7 |
tqdm==4.61.2
|
| 8 |
opencv-python
|
| 9 |
-
--find-links https://download.pytorch.org/whl/
|
| 10 |
-
torch==1.
|
|
|
|
| 6 |
timm==0.5.4
|
| 7 |
tqdm==4.61.2
|
| 8 |
opencv-python
|
| 9 |
+
--find-links https://download.pytorch.org/whl/torch_stable.html
|
| 10 |
+
torch==1.5.0+gpu
|