From cbf2806de05d107bdfccb61a76bf25e780d3482a Mon Sep 17 00:00:00 2001 From: Vijaya Date: Thu, 29 May 2025 17:17:47 +0000 Subject: [PATCH] Add gcloud setup same as Maxtext --- experimental/jetstream-maxtext-stable-stack/Dockerfile | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/experimental/jetstream-maxtext-stable-stack/Dockerfile b/experimental/jetstream-maxtext-stable-stack/Dockerfile index ed2209ea..dba28797 100644 --- a/experimental/jetstream-maxtext-stable-stack/Dockerfile +++ b/experimental/jetstream-maxtext-stable-stack/Dockerfile @@ -64,6 +64,16 @@ RUN apt-get update \ RUN python3 -m pip install --upgrade pip +# Add the Google Cloud SDK package repository +RUN echo "deb [signed-by=/usr/share/keyrings/cloud.google.gpg] https://packages.cloud.google.com/apt cloud-sdk main" | tee -a /etc/apt/sources.list.d/google-cloud-sdk.list +RUN curl https://packages.cloud.google.com/apt/doc/apt-key.gpg | apt-key --keyring /usr/share/keyrings/cloud.google.gpg add - + +# Install the Google Cloud SDK +RUN apt-get update && apt-get install -y google-cloud-sdk + +# Set environment variables for Google Cloud SDK +ENV PATH="/usr/local/google-cloud-sdk/bin:${PATH}" + # Install MaxText package COPY --from=maxtext_cloner /src . RUN cd maxtext && bash setup.sh