Update pip in JetStream Pytorch and checkpoint Dockerfiles (#739)

GoogleCloudPlatform · Jul 15, 2024 · 6d21def · 6d21def
1 parent 0fd14fd
commit 6d21def
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tutorials-and-examples/inference-servers/checkpoints/Dockerfile b/tutorials-and-examples/inference-servers/checkpoints/Dockerfile
@@ -13,6 +13,8 @@ RUN apt -y update && apt install -y --no-install-recommends \
     curl \
     gnupg
 
+RUN python3 -m pip install --upgrade pip
+
 RUN update-alternatives --install \
     /usr/bin/python3 python3 /usr/bin/python3.10 1
 

diff --git a/...ials-and-examples/inference-servers/jetstream/pytorch/jetstream-pytorch-server/Dockerfile b/...ials-and-examples/inference-servers/jetstream/pytorch/jetstream-pytorch-server/Dockerfile
@@ -12,6 +12,8 @@ RUN apt -y update && apt install -y --no-install-recommends \
     python3.10 \
     python3-pip
 
+RUN python3 -m pip install --upgrade pip
+
 RUN update-alternatives --install \
     /usr/bin/python3 python3 /usr/bin/python3.10 1