requirements
Browse files- requirements.txt +15 -15
requirements.txt
CHANGED
|
@@ -1,18 +1,18 @@
|
|
| 1 |
-
--extra-index-url https://download.pytorch.org/whl/
|
| 2 |
-
nvidia-cublas-
|
| 3 |
-
nvidia-cuda-cupti-
|
| 4 |
-
nvidia-cuda-nvrtc-
|
| 5 |
-
nvidia-cuda-runtime-
|
| 6 |
-
nvidia-cudnn-
|
| 7 |
-
nvidia-cufft-
|
| 8 |
-
nvidia-curand-
|
| 9 |
-
nvidia-cusolver-
|
| 10 |
-
nvidia-cusparse-
|
| 11 |
-
nvidia-cusparselt-
|
| 12 |
-
nvidia-nccl-
|
| 13 |
-
nvidia-nvjitlink-
|
| 14 |
-
nvidia-nvtx-
|
| 15 |
-
cupy-
|
| 16 |
huggingface-hub
|
| 17 |
kornia
|
| 18 |
kornia_rs
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu118
|
| 2 |
+
nvidia-cublas-cu11
|
| 3 |
+
nvidia-cuda-cupti-cu11
|
| 4 |
+
nvidia-cuda-nvrtc-cu11
|
| 5 |
+
nvidia-cuda-runtime-cu11
|
| 6 |
+
nvidia-cudnn-cu11
|
| 7 |
+
nvidia-cufft-cu11
|
| 8 |
+
nvidia-curand-cu11
|
| 9 |
+
nvidia-cusolver-cu11
|
| 10 |
+
nvidia-cusparse-cu11
|
| 11 |
+
nvidia-cusparselt-cu11
|
| 12 |
+
nvidia-nccl-cu11
|
| 13 |
+
nvidia-nvjitlink-cu11
|
| 14 |
+
nvidia-nvtx-cu11
|
| 15 |
+
cupy-cuda11x
|
| 16 |
huggingface-hub
|
| 17 |
kornia
|
| 18 |
kornia_rs
|