Skip to content

Commit b4c62d1

Browse files
Separate GPU accelaration dependencies
1 parent 14dd5c1 commit b4c62d1

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

requirements_gpu.txt

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
nvidia-cublas-cu12==12.4.5.8
2+
nvidia-cuda-cupti-cu12==12.4.127
3+
nvidia-cuda-nvrtc-cu12==12.4.127
4+
nvidia-cuda-runtime-cu12==12.4.127
5+
nvidia-cudnn-cu12==9.1.0.70
6+
nvidia-cufft-cu12==11.2.1.3
7+
nvidia-curand-cu12==10.3.5.147
8+
nvidia-cusolver-cu12==11.6.1.9
9+
nvidia-cusparse-cu12==12.3.1.170
10+
nvidia-nccl-cu12==2.21.5
11+
nvidia-nvjitlink-cu12==12.4.127
12+
nvidia-nvtx-cu12==12.4.127
13+
triton==3.1.0

0 commit comments

Comments
 (0)