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

Updated minimum jaxlib to 0.1.51 (#3642)

Browse Source
This commit is contained in:
George Necula 2020-07-02 18:53:58 +03:00 committed by GitHub
parent bfe8f8732b
commit 166e795d63
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -410,7 +410,7 @@ PYTHON_VERSION=cp37 # alternatives: cp36, cp37, cp38
CUDA_VERSION=cuda100 # alternatives: cuda100, cuda101, cuda102, cuda110
PLATFORM=manylinux2010_x86_64 # alternatives: manylinux2010_x86_64
BASE_URL='https://storage.googleapis.com/jax-releases'
pip install --upgrade $BASE_URL/$CUDA_VERSION/jaxlib-0.1.50-$PYTHON_VERSION-none-$PLATFORM.whl
pip install --upgrade $BASE_URL/$CUDA_VERSION/jaxlib-0.1.51-$PYTHON_VERSION-none-$PLATFORM.whl
pip install --upgrade jax # install jax
```
@ -448,7 +448,7 @@ requires Python 3.6 or above. Jax does not support Python 2 any more.
To try automatic detection of the correct version for your system, you can run:
```bash
pip install --upgrade https://storage.googleapis.com/jax-releases/`nvidia-smi | sed -En "s/.* CUDA Version: ([0-9]*)\.([0-9]*).*/cuda\1\2/p"`/jaxlib-0.1.50-`python3 -V | sed -En "s/Python ([0-9]*)\.([0-9]*).*/cp\1\2/p"`-none-manylinux2010_x86_64.whl jax
pip install --upgrade https://storage.googleapis.com/jax-releases/`nvidia-smi | sed -En "s/.* CUDA Version: ([0-9]*)\.([0-9]*).*/cuda\1\2/p"`/jaxlib-0.1.51-`python3 -V | sed -En "s/Python ([0-9]*)\.([0-9]*).*/cp\1\2/p"`-none-manylinux2010_x86_64.whl jax
```
Please let us know on [the issue tracker](https://github.com/google/jax/issues)

2
build/test-requirements.txt
View File

@ -1,5 +1,5 @@
flake8
jaxlib==0.1.48
jaxlib==0.1.51
msgpack
mypy==0.770
pytest-benchmark

2
jax/lib/__init__.py
View File

@ -22,7 +22,7 @@ __all__ = [
import jaxlib
_minimum_jaxlib_version = (0, 1, 48)
_minimum_jaxlib_version = (0, 1, 51)
try:
from jaxlib import version as jaxlib_version
except Exception as err: