Skip to content

Commit 822bc36

Browse files
authored
Update quickstart to suggested jax[cuda12] installation
This PR updates the quickstart docs to match the README and Installation Guide suggestion of using ``` pip install jax[cuda12] ``` for GPU.
1 parent 8fec5d6 commit 822bc36

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/quickstart.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ pip install "jax[cpu]"
3131
```
3232
or, for NVIDIA GPU:
3333
```
34-
pip install -U "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
34+
pip install -U "jax[cuda12]"
3535
```
3636
For more detailed platform-specific installation information, check out {ref}`installation`.
3737

0 commit comments

Comments
 (0)