Add CUDA 12 cuBLAS dependency
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -3,6 +3,7 @@ spaces>=0.41.0
|
|
| 3 |
numpy
|
| 4 |
tqdm
|
| 5 |
msgpack
|
|
|
|
| 6 |
nvidia-cuda-nvrtc-cu12>=12.4
|
| 7 |
nvidia-cuda-runtime-cu12>=12.4
|
| 8 |
sgl_kernel>=0.3.17.post1
|
|
|
|
| 3 |
numpy
|
| 4 |
tqdm
|
| 5 |
msgpack
|
| 6 |
+
nvidia-cublas-cu12>=12.4
|
| 7 |
nvidia-cuda-nvrtc-cu12>=12.4
|
| 8 |
nvidia-cuda-runtime-cu12>=12.4
|
| 9 |
sgl_kernel>=0.3.17.post1
|