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

update jaxlib wheel urls to 0.1.1

Browse Source
This commit is contained in:
Matthew Johnson 2018-12-13 17:35:19 -08:00
parent e4ef7eae27
commit 2e40e63527
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -142,7 +142,7 @@ cloud VM), you can run
PYTHON_VERSION=py2 # alternatives: py2, py3
CUDA_VERSION=cuda92 # alternatives: cuda90, cuda92, cuda100
PLATFORM=linux_x86_64 # alternatives: linux_x86_64
pip install https://storage.googleapis.com/jax-wheels/$CUDA_VERSION/jaxlib-0.1-$PYTHON_VERSION-none-$PLATFORM.whl
pip install https://storage.googleapis.com/jax-wheels/$CUDA_VERSION/jaxlib-0.1.1-$PYTHON_VERSION-none-$PLATFORM.whl
pip install jax # install jax
```

2
notebooks/gufuncs.ipynb
View File

@ -125,7 +125,7 @@
},
"cell_type": "code",
"source": [
"!pip install -q https://storage.googleapis.com/jax-wheels/cuda92/jaxlib-0.1-py3-none-any.whl\n",
"!pip install -q https://storage.googleapis.com/jax-wheels/cuda92/jaxlib-0.1.1-py3-none-any.whl\n",
"!pip install --upgrade -q jax"
],
"execution_count": 0,

2
notebooks/neural_network_and_data_loading.ipynb
View File

@ -74,7 +74,7 @@
},
"cell_type": "code",
"source": [
"!pip install https://storage.googleapis.com/jax-wheels/cuda92/jaxlib-0.1-py3-none-any.whl\n",
"!pip install https://storage.googleapis.com/jax-wheels/cuda92/jaxlib-0.1.1-py3-none-any.whl\n",
"!pip install --upgrade jax"
],
"execution_count": 0,

2
notebooks/quickstart.ipynb
View File

@ -89,7 +89,7 @@
},
"cell_type": "code",
"source": [
"!pip install https://storage.googleapis.com/jax-wheels/cuda92/jaxlib-0.1-py3-none-any.whl\n",
"!pip install https://storage.googleapis.com/jax-wheels/cuda92/jaxlib-0.1.1-py3-none-any.whl\n",
"!pip install --upgrade jax"
],
"execution_count": 0,