feat: add in pytorch base image

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" ]

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