2025-03-11 08:29:45 -07:00
|
|
|
# NVIDIA CUDA dependencies
|
|
|
|
|
# Note that the wheels are downloaded only when the targets in bazel command
|
|
|
|
|
# contain dependencies on these wheels.
|
2025-03-11 12:32:58 -07:00
|
|
|
nvidia-cublas-cu12>=12.1.3.1 ; sys_platform == "linux"
|
|
|
|
|
nvidia-cuda-cupti-cu12>=12.1.105 ; sys_platform == "linux"
|
|
|
|
|
nvidia-cuda-nvcc-cu12>=12.6.85 ; sys_platform == "linux"
|
|
|
|
|
nvidia-cuda-runtime-cu12>=12.1.105 ; sys_platform == "linux"
|
2025-03-27 09:53:09 -07:00
|
|
|
nvidia-cudnn-cu12>=9.8,<10.0 ; sys_platform == "linux"
|
2025-03-11 12:32:58 -07:00
|
|
|
nvidia-cufft-cu12>=11.0.2.54 ; sys_platform == "linux"
|
|
|
|
|
nvidia-cusolver-cu12>=11.4.5.107 ; sys_platform == "linux"
|
|
|
|
|
nvidia-cusparse-cu12>=12.1.0.106 ; sys_platform == "linux"
|
|
|
|
|
nvidia-nccl-cu12>=2.18.1 ; sys_platform == "linux"
|
|
|
|
|
nvidia-nvjitlink-cu12>=12.1.105 ; sys_platform == "linux"
|
