mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update README to drop CUDA 9.0 and add CUDA 10.2

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2020-01-29 11:23:27 -08:00
parent 991324f8df
commit 633aa00392
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -407,7 +407,7 @@ cloud VM), you can run
```bash
# install jaxlib
PYTHON_VERSION=cp37 # alternatives: cp35, cp36, cp37, cp38
CUDA_VERSION=cuda92 # alternatives: cuda90, cuda92, cuda100, cuda101
CUDA_VERSION=cuda92 # alternatives: cuda92, cuda100, cuda101, cuda102
PLATFORM=linux_x86_64 # alternatives: linux_x86_64
BASE_URL='https://storage.googleapis.com/jax-releases'
pip install --upgrade $BASE_URL/$CUDA_VERSION/jaxlib-0.1.38-$PYTHON_VERSION-none-$PLATFORM.whl
@ -416,8 +416,8 @@ pip install --upgrade jax # install jax
```
The library package name must correspond to the version of the existing CUDA
installation you want to use, with `cuda101` for CUDA 10.1, `cuda100` for CUDA
10.0, `cuda92` for CUDA 9.2, and `cuda90` for CUDA 9.0. To find your CUDA and
installation you want to use, with `cuda102` for CUDA 10.2, `cuda101` for CUDA
10.1, `cuda100` for CUDA 10.0, and `cuda92` for CUDA 9.2. To find your CUDA and
CUDNN versions, you can run commands like these, depending on your CUDNN install
path: