From e62e4683b68e29c0aa0abbaab827f9d388415e8c Mon Sep 17 00:00:00 2001
From: ahodges22 <ahodges@articulate.com>
Date: Wed, 12 Jun 2024 16:48:36 -0700
Subject: [PATCH] feat: add in pytorch base image

---
 3.12-pytorch/Dockerfile | 40 ++++++++++++++++++++++++++++++++++++++++
 README.md               |  1 +
 2 files changed, 41 insertions(+)
 create mode 100644 3.12-pytorch/Dockerfile

diff --git a/3.12-pytorch/Dockerfile b/3.12-pytorch/Dockerfile
new file mode 100644
index 0000000..f0abb6b
--- /dev/null
+++ b/3.12-pytorch/Dockerfile
@@ -0,0 +1,40 @@
+# tags=articulate/python:3.12-pytorch
+# syntax=docker/dockerfile:1
+FROM python:3.12-slim-bookworm
+
+ENV SERVICE_ROOT /service
+ENV SERVICE_USER service
+ENV SERVICE_UID 1001
+
+ARG TARGETARCH
+
+ADD --chmod=755 https://raw.githubusercontent.com/articulate/docker-bootstrap/main/scripts/install_packages /usr/local/bin/install_packages
+ADD --chmod=755 https://raw.githubusercontent.com/articulate/docker-bootstrap/main/scripts/awscli.sh /tmp/awscli.sh
+
+RUN install_packages make dumb-init && /tmp/awscli.sh && rm /tmp/awscli.sh \
+    && groupadd --gid $SERVICE_UID $SERVICE_USER \
+    && useradd --create-home --shell /bin/bash --gid $SERVICE_UID --uid $SERVICE_UID $SERVICE_USER
+
+ADD --chmod=755 https://github.com/articulate/docker-bootstrap/releases/latest/download/docker-bootstrap_linux_${TARGETARCH} /entrypoint
+ADD --chmod=755 https://raw.githubusercontent.com/articulate/docker-bootstrap/main/scripts/docker-secrets /usr/local/bin/secrets
+ADD --chmod=755 https://raw.githubusercontent.com/vishnubob/wait-for-it/81b1373f17855a4dc21156cfe1694c31d7d1792e/wait-for-it.sh /wait-for-it.sh
+
+# Environment variables for PyTorch version and CUDA version
+# Set build arguments for PyTorch version
+ENV PYTORCH_VERSION=2.3.1
+ENV TORCHVISION_VERSION=0.18.1
+ENV CUDA_VERSION=12.1
+
+# Install PyTorch
+# Added to satisfy linter
+SHELL ["/bin/bash", "-o", "pipefail", "-c"]
+RUN CUDA_VERSION_SANITIZED="$(echo $CUDA_VERSION | tr -d '.')" \
+    && pip install --no-cache-dir torch=="${PYTORCH_VERSION}" torchaudio=="${PYTORCH_VERSION}" torchvision=="${TORCHVISION_VERSION}" --index-url "https://download.pytorch.org/whl/cu${CUDA_VERSION_SANITIZED}"
+
+USER $SERVICE_USER
+WORKDIR $SERVICE_ROOT
+
+# Our entrypoint will pull in our environment variables from Consul and Vault,
+# and execute whatever command we provided the container.
+# See https://github.com/articulate/docker-bootstrap
+ENTRYPOINT [ "dumb-init", "--", "/entrypoint" ]
diff --git a/README.md b/README.md
index 6470259..2f55a6f 100644
--- a/README.md
+++ b/README.md
@@ -18,6 +18,7 @@ Base Python Docker images.
 > 🌟 recommended image
 
 * __articulate/python:3.12__ 🌟
+* articulate/python:3.12-pytorch
 * articulate/python:3.11
 
 ## Creating a new image