mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00
14 lines
713 B
Plaintext
14 lines
713 B
Plaintext
# 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"
|