mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update hermetic CUDA docs.

Browse Source 
PiperOrigin-RevId: 689463215
This commit is contained in:
jax authors 2024-10-24 11:50:24 -07:00
parent bd417ba6d0
commit 8c9dc21e30
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
docs/developer.md
View File

@ -109,6 +109,11 @@ current directory.
--cuda_version=12.3.2 --cudnn_version=9.1.1
```
Please note that these parameters are optional: by default Bazel will
download CUDA and CUDNN redistribution versions provided in `.bazelrc` in the
environment variables `HERMETIC_CUDA_VERSION` and `HERMETIC_CUDNN_VERSION`
respectively.
To point to CUDA/CUDNN/NCCL redistributions on local file system, you can use
the following command: