File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -452,7 +452,7 @@ Next, run
452
452
pip install --upgrade pip
453
453
# Installs the wheel compatible with CUDA 11 and cuDNN 8.2 or newer.
454
454
# 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
456
456
```
457
457
458
458
@@ -464,10 +464,10 @@ version for jaxlib explicitly:
464
464
pip install --upgrade pip
465
465
466
466
# 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
468
468
469
469
# 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
471
471
```
472
472
473
473
You can find your CUDA version with the command:
You can’t perform that action at this time.
0 commit comments