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

DOC: update installation instructions for compatibility with zsh

Browse Source
This commit is contained in:
Jake VanderPlas 2021-06-28 12:57:24 -07:00
parent 0d68dbd619
commit 5e2681939a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -403,7 +403,7 @@ development on a laptop, you can run
```bash
pip install --upgrade pip
pip install --upgrade jax[cpu]
pip install --upgrade "jax[cpu]"
```
On Linux, it is often necessary to first update `pip` to a version that supports
@ -424,7 +424,7 @@ Next, run
```bash
pip install --upgrade pip
pip install --upgrade jax[cuda111] -f https://storage.googleapis.com/jax-releases/jax_releases.html
pip install --upgrade "jax[cuda111]" -f https://storage.googleapis.com/jax-releases/jax_releases.html
```
The jaxlib version must correspond to the version of the existing CUDA