fix
Browse files- requirements.txt +13 -13
requirements.txt
CHANGED
|
@@ -1,4 +1,17 @@
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu124
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
cupy-cuda12x
|
| 3 |
huggingface-hub
|
| 4 |
kornia
|
|
@@ -13,16 +26,3 @@ torchvision==0.19.0
|
|
| 13 |
torchaudio==2.4.0
|
| 14 |
tqdm
|
| 15 |
triton
|
| 16 |
-
nvidia-cublas-cu12
|
| 17 |
-
nvidia-cuda-cupti-cu12
|
| 18 |
-
nvidia-cuda-nvrtc-cu12
|
| 19 |
-
nvidia-cuda-runtime-cu12
|
| 20 |
-
nvidia-cudnn-cu12
|
| 21 |
-
nvidia-cufft-cu12
|
| 22 |
-
nvidia-curand-cu12
|
| 23 |
-
nvidia-cusolver-cu12
|
| 24 |
-
nvidia-cusparse-cu12
|
| 25 |
-
nvidia-cusparselt-cu12
|
| 26 |
-
nvidia-nccl-cu12
|
| 27 |
-
nvidia-nvjitlink-cu12
|
| 28 |
-
nvidia-nvtx-cu12
|
|
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu124
|
| 2 |
+
nvidia-cublas-cu12
|
| 3 |
+
nvidia-cuda-cupti-cu12
|
| 4 |
+
nvidia-cuda-nvrtc-cu12
|
| 5 |
+
nvidia-cuda-runtime-cu12
|
| 6 |
+
nvidia-cudnn-cu12
|
| 7 |
+
nvidia-cufft-cu12
|
| 8 |
+
nvidia-curand-cu12
|
| 9 |
+
nvidia-cusolver-cu12
|
| 10 |
+
nvidia-cusparse-cu12
|
| 11 |
+
nvidia-cusparselt-cu12
|
| 12 |
+
nvidia-nccl-cu12
|
| 13 |
+
nvidia-nvjitlink-cu12
|
| 14 |
+
nvidia-nvtx-cu12
|
| 15 |
cupy-cuda12x
|
| 16 |
huggingface-hub
|
| 17 |
kornia
|
|
|
|
| 26 |
torchaudio==2.4.0
|
| 27 |
tqdm
|
| 28 |
triton
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|