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

Fix #24661: Add zsh support to conda install documentation

Browse Source
This commit is contained in:
Li-Jesse-Jiaze 2024-11-01 17:57:18 +01:00
parent 24f3a2bb79
commit 5e1366c4ce
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/installation.md
View File

@ -244,7 +244,7 @@ conda install jax -c conda-forge
To install it on a machine with an NVIDIA GPU, run:
```bash
conda install jaxlib=*=*cuda* jax cuda-nvcc -c conda-forge -c nvidia
conda install "jaxlib=*=*cuda*" jax cuda-nvcc -c conda-forge -c nvidia
```
Note the `cudatoolkit` distributed by `conda-forge` is missing `ptxas`, which