chore: deprecate p39 docker builds (#678)

ZettaAI · Apr 25, 2024 · 295c10d · 295c10d
1 parent f5d4f6d
commit 295c10d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 11 deletions.
diff --git a/build_image.py b/build_image.py
@@ -23,8 +23,8 @@ def main():
         "--python",
         "-p",
         type=str,
-        default="3.9",
-        choices=["3.9", "3.10"],
+        default="3.10",
+        choices=["3.11", "3.10"],
         help="Which python version to use for the image.",
     )
     parser.add_argument("--region", type=str, default="us-east1", help="Artifact Registry region.")

diff --git a/docker/Dockerfile.all.p39 → docker/Dockerfile.all.p311 b/docker/Dockerfile.all.p39 → docker/Dockerfile.all.p311
@@ -1,10 +1,18 @@
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+#FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
+FROM nvidia/cuda:12.2.0-runtime-ubuntu20.04
 
 ENV DEBIAN_FRONTEND="noninteractive"
 
 RUN apt-get update \
   && apt-get install -y git build-essential wget curl vim ffmpeg libsm6 libxext6 software-properties-common unixodbc-dev \
-  && pip install --no-cache-dir posix-ipc gevent \
+  && add-apt-repository ppa:deadsnakes/ppa -y \
+  && apt-get update \
+  && apt-get install -y --no-install-recommends python3.11 python3-pip  python3.11-dev \
+  && update-alternatives --install /usr/bin/python python /usr/bin/python3.11 1 \
+  && apt-get install -y python3.11-distutils \
+  && curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py \
+  && python3.11 get-pip.py \
+  && pip install posix-ipc \
   && apt-get --purge autoremove -y build-essential \
   && apt-get clean \
   && rm -rf /var/lib/apt/lists/* \
@@ -13,15 +21,10 @@ RUN apt-get update \
   && wget https://github.com/cue-lang/cue/releases/download/v0.4.3/cue_v0.4.3_linux_amd64.tar.gz \
   && tar -xzvf cue_v0.4.3_linux_amd64.tar.gz  \
   && cp cue /bin/
-#&& add-apt-repository ppa:deadsnakes/ppa -y \
-  #&& apt-get install -y --no-install-recommends python3.9 python3-pip  python3.9-dev \
-  #&& update-alternatives --install /usr/bin/python python /usr/bin/python3.9 1 \
-  #&& apt-get install -y python3.9-distutils \
-  #&& curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py \
-  #&& python3.9 get-pip.py \
+
 ENV PYTHONPATH /opt/zetta_utils
 WORKDIR /opt/zetta_utils
 ADD pyproject.toml /opt/zetta_utils/
-RUN pip install --no-cache-dir '.[modules]'
+RUN pip install '.[modules]'
 COPY . /opt/zetta_utils/
 RUN zetta --help