New URL for CUDA

Releases URL has changed: jax-ml/jax#11142
e-bacho · Jul 14, 2022 · a2f8bf0 · a2f8bf0
1 parent ec07a90
commit a2f8bf0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Docker/Dockerfile b/Docker/Dockerfile
@@ -6,7 +6,7 @@ RUN apt-get update && apt-get install -y \
   python3-pip \
   && rm -rf /var/lib/apt/lists/*
 
-RUN pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_releases.html \
+RUN pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html \
   && pip install -q \
   git+https://github.com/borisdayma/dalle-mini.git \
   git+https://github.com/patil-suraj/vqgan-jax.git