chore(components): Update Dockerfile

Signed-off-by: Ze Mao <zemao@google.com> PiperOrigin-RevId: 700828757
kubeflow · Nov 28, 2024 · ca004cc · ca004cc
1 parent 3727961
commit ca004cc
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/components/google-cloud/Dockerfile b/components/google-cloud/Dockerfile
@@ -16,18 +16,11 @@
 FROM marketplace.gcr.io/google/ubuntu2404:latest
 RUN apt update && apt -y install git python3 \
   python3-pip \
-  python3-venv
+  python3-setuptools
 
 WORKDIR /root
 
-# Create a virtual environment
-RUN python3 -m venv venv
-
-# Activate the virtual environment
-ENV PATH="venv/bin:$PATH"
-
-# Python 3.12 removed distutils
-RUN pip3 install -U pip setuptools
+ENV PIP_BREAK_SYSTEM_PACKAGES=1
 
 # Required by gcp_launcher
 # Using google-cloud-aiplatform>=1.21.0 to avoid dataset creatation timeout