Fix Dockerfile breakage due to google/jax#11142.

PiperOrigin-RevId: 456029742 Change-Id: I908c5fe07d14168d65fccc3e42cb5c994200b0e2

Fix Dockerfile breakage due to google/jax#11142.
PiperOrigin-RevId: 456029742 Change-Id: I908c5fe07d14168d65fccc3e42cb5c994200b0e2
624a4496 · Tim Green · Copybara-Service · 7b2a4c5a · 624a4496
Commit 624a4496 authored Jun 20, 2022 by Tim Green Committed by Copybara-Service Jun 20, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

docker/Dockerfile docker/Dockerfile +1 -1

No files found.
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -68,7 +68,7 @@ RUN pip3 install --upgrade pip \
    && pip3 install --upgrade \
      jax==0.2.14 \
      jaxlib==0.1.69+cuda$(cut -f1,2 -d. <<< ${CUDA} | sed 's/\.//g') \
-      -f https://storage.googleapis.com/jax-releases/jax_releases.html
+      -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

 # Apply OpenMM patch.
 WORKDIR /opt/conda/lib/python3.7/site-packages