# NVIDIA CUDA dependencies
# Note that the wheels are downloaded only when the targets in bazel command
# contain dependencies on these wheels.
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"
nvidia-cudnn-cu12>=9.1,<10.0 ; sys_platform == "linux"
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"