diff --git a/README.md b/README.md index 80f030991..d88414d5f 100644 --- a/README.md +++ b/README.md @@ -131,7 +131,7 @@ PYTHON_VERSION=cp37 # alternatives: cp27, cp35, cp36, cp37 CUDA_VERSION=cuda92 # alternatives: cuda90, cuda92, cuda100, cuda101 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.33-$PYTHON_VERSION-none-$PLATFORM.whl +pip install --upgrade $BASE_URL/$CUDA_VERSION/jaxlib-0.1.34-$PYTHON_VERSION-none-$PLATFORM.whl pip install --upgrade jax # install jax ```