Fix Dockerfile breakage due to jax-ml/jax#11142.

PiperOrigin-RevId: 456029742 Change-Id: I908c5fe07d14168d65fccc3e42cb5c994200b0e2
dogrecajb · Jun 20, 2022 · 0bf65b1 · 0bf65b1
1 parent 1744087
commit 0bf65b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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