requ
Browse files- requirements.txt +13 -0
requirements.txt
CHANGED
|
@@ -13,3 +13,16 @@ torchvision==0.19.0
|
|
| 13 |
torchaudio==2.4.0
|
| 14 |
tqdm
|
| 15 |
triton
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|