You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
445 B
Plaintext
22 lines
445 B
Plaintext
# Giant ML and CUDA libraries (stable - rarely changed)
|
|
accelerate==1.10.1
|
|
cupy-cuda12x==13.6.0
|
|
nvidia-cublas-cu12
|
|
nvidia-cuda-cupti-cu12==12.4.127
|
|
nvidia-cuda-runtime-cu12==12.4.127
|
|
nvidia-cudnn-cu12
|
|
nvidia-cufft-cu12
|
|
nvidia-cufile-cu12
|
|
nvidia-curand-cu12
|
|
nvidia-cusolver-cu12
|
|
nvidia-cusparse-cu12
|
|
nvidia-cusparselt-cu12
|
|
nvidia-nccl-cu12
|
|
nvidia-nvjitlink-cu12
|
|
nvidia-nvtx-cu12
|
|
torch==2.6.0
|
|
torchaudio==2.6.0
|
|
torchvision==0.21.0
|
|
triton
|
|
xformers
|