Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -16,4 +16,5 @@ pyyaml>3.10
|
|
16 |
colorlog
|
17 |
plotly-express
|
18 |
setuptools
|
19 |
-
nvidia-cuda-nvcc-cu11
|
|
|
|
16 |
colorlog
|
17 |
plotly-express
|
18 |
setuptools
|
19 |
+
nvidia-cuda-nvcc-cu11
|
20 |
+
--extra-index-url https://download.pytorch.org/whl/cu118
|