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

Update jaxlib release to 0.1.23.

Browse Source
This commit is contained in:
Peter Hawkins 2019-08-08 22:09:30 -04:00
parent 95b8bea50b
commit 1b672dc28d
9 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -163,8 +163,8 @@ cloud VM), you can run
PYTHON_VERSION=cp27 # alternatives: cp27, cp35, cp36, cp37
CUDA_VERSION=cuda92 # alternatives: cuda90, cuda92, cuda100
PLATFORM=linux_x86_64 # alternatives: linux_x86_64
BASE_URL='https://storage.googleapis.com/jax-wheels'
pip install --upgrade $BASE_URL/$CUDA_VERSION/jaxlib-0.1.21-$PYTHON_VERSION-none-$PLATFORM.whl
BASE_URL='https://storage.googleapis.com/jax-releases'
pip install --upgrade $BASE_URL/$CUDA_VERSION/jaxlib-0.1.23-$PYTHON_VERSION-none-$PLATFORM.whl
pip install --upgrade jax # install jax
```

2
notebooks/XLA_in_Python.ipynb
View File

@ -110,7 +110,7 @@
},
"outputs": [],
"source": [
"!pip install --upgrade -q https://storage.googleapis.com/jax-wheels/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.21-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q https://storage.googleapis.com/jax-releases/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.23-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q jax"
]
},

2
notebooks/gufuncs.ipynb
View File

@ -125,7 +125,7 @@
"colab": {}
},
"source": [
"!pip install --upgrade -q https://storage.googleapis.com/jax-wheels/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.21-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q https://storage.googleapis.com/jax-releases/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.23-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q jax"
],
"execution_count": 0,

4
notebooks/maml.ipynb
View File

@ -63,7 +63,7 @@
"colab": {}
},
"source": [
"!pip install --upgrade -q https://storage.googleapis.com/jax-wheels/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.21-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q https://storage.googleapis.com/jax-releases/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.23-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q jax"
],
"execution_count": 0,
@ -912,4 +912,4 @@
"outputs": []
}
]
}
}

2
notebooks/neural_network_and_data_loading.ipynb
View File

@ -74,7 +74,7 @@
},
"cell_type": "code",
"source": [
"!pip install --upgrade -q https://storage.googleapis.com/jax-wheels/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.21-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q https://storage.googleapis.com/jax-releases/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.23-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q jax"
],
"execution_count": 0,

2
notebooks/neural_network_with_tfds_data.ipynb
View File

@ -87,7 +87,7 @@
}
],
"source": [
"!pip install --upgrade -q https://storage.googleapis.com/jax-wheels/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.21-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q https://storage.googleapis.com/jax-releases/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.23-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q jax"
]
},

2
notebooks/quickstart.ipynb
View File

@ -89,7 +89,7 @@
"colab": {}
},
"source": [
"!pip install --upgrade -q https://storage.googleapis.com/jax-wheels/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.21-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q https://storage.googleapis.com/jax-releases/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.23-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q jax"
],
"execution_count": 0,

2
notebooks/score_matching.ipynb
View File

@ -50,7 +50,7 @@
"source": [
"# but first, let's install jax\n",
"\n",
"!pip install --upgrade -q https://storage.googleapis.com/jax-wheels/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.22-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q https://storage.googleapis.com/jax-releases/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.23-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q jax"
]
},

2
notebooks/vmapped log-probs.ipynb
View File

@ -22,7 +22,7 @@
},
"cell_type": "code",
"source": [
"!pip install --upgrade -q https://storage.googleapis.com/jax-wheels/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.21-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q https://storage.googleapis.com/jax-releases/cuda$(echo $CUDA_VERSION | sed -e 's/\\.//' -e 's/\\..*//')/jaxlib-0.1.23-cp36-none-linux_x86_64.whl\n",
"!pip install --upgrade -q jax"
],
"execution_count": 0,