Commit f4a1be7
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -50,7 +50,7 @@ ENV JAX_CUDA_VERSION=cuda100 | |||
| 50 | 50 | ENV JAX_PLATFORM=linux_x86_64 | |
| 51 | 51 | ENV JAX_BASE_URL="https://storage.googleapis.com/jax-releases" | |
| 52 | 52 | ||
| 53 | - RUN pip install --upgrade $JAX_BASE_URL/$JAX_CUDA_VERSION/jaxlib-0.1.36-$JAX_PYTHON_VERSION-none-$JAX_PLATFORM.whl && \ | ||
| 53 | + RUN pip install --upgrade $JAX_BASE_URL/$JAX_CUDA_VERSION/jaxlib-0.1.37-$JAX_PYTHON_VERSION-none-$JAX_PLATFORM.whl && \ | ||
| 54 | 54 | pip install --upgrade jax | |
| 55 | 55 | ||
| 56 | 56 | # Reinstall packages with a separate version for GPU support. | |
0 commit comments