mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
Update README.md
This commit is contained in:
parent
0cd84efac1
commit
9f61472a91
@ -123,7 +123,7 @@ To install a CPU-only version, which might be useful for doing local
|
||||
development on a laptop, you can run
|
||||
|
||||
```bash
|
||||
pip install jax jaxlib
|
||||
pip install jax jaxlib # CPU-only version
|
||||
```
|
||||
|
||||
If you want to install JAX with both CPU and GPU support, using existing CUDA
|
||||
|
Loading…
x
Reference in New Issue
Block a user