Skip to content

Commit e780a40

Browse files
yashk2810jax authors
authored andcommitted
Split cuda and nocuda index files into their own
PiperOrigin-RevId: 455463113
1 parent 6375515 commit e780a40

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -452,7 +452,7 @@ Next, run
452452
pip install --upgrade pip
453453
# Installs the wheel compatible with CUDA 11 and cuDNN 8.2 or newer.
454454
# Note: wheels only available on linux.
455-
pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_releases.html
455+
pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
456456
```
457457

458458

@@ -464,10 +464,10 @@ version for jaxlib explicitly:
464464
pip install --upgrade pip
465465

466466
# Installs the wheel compatible with Cuda >= 11.4 and cudnn >= 8.2
467-
pip install "jax[cuda11_cudnn82]" -f https://storage.googleapis.com/jax-releases/jax_releases.html
467+
pip install "jax[cuda11_cudnn82]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
468468

469469
# Installs the wheel compatible with Cuda >= 11.1 and cudnn >= 8.0.5
470-
pip install "jax[cuda11_cudnn805]" -f https://storage.googleapis.com/jax-releases/jax_releases.html
470+
pip install "jax[cuda11_cudnn805]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
471471
```
472472

473473
You can find your CUDA version with the command:

0 commit comments

Comments
 (0)