remove libtpu pypi location

apple · Jan 22, 2025 · 6bc7cdf · 6bc7cdf
1 parent 4a6ada5
commit 6bc7cdf
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -146,8 +146,7 @@ RUN apt update -y && apt-get install -y google-perftools glibc-tools
 ENV PIP_FIND_LINKS=https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 RUN pip install .[core,gpu]
 COPY . .
-RUN pip install -U "jax[gpu]==0.4.38" "jax==0.4.38" "jaxlib==0.4.38" \
-    -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
+RUN pip install -U "jax[gpu]==0.4.38" "jax==0.4.38" "jaxlib==0.4.38"
 COPY . .