mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump jaxlib version in README to 0.1.39

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2020-02-14 10:06:27 -08:00
parent 6d0d6fd6c7
commit 2a6e60fd50
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -410,7 +410,7 @@ PYTHON_VERSION=cp37 # alternatives: cp35, cp36, cp37, cp38
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
pip install --upgrade $BASE_URL/$CUDA_VERSION/jaxlib-0.1.39-$PYTHON_VERSION-none-$PLATFORM.whl
pip install --upgrade jax # install jax
```