feat: add in pytorch base image

3.12-pytorch/Dockerfile

@@ -0,0 +1,17 @@
+FROM articulate/python:3.12
+
+# 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
+RUN CUDA_VERSION_SANITIZED=$(echo $CUDA_VERSION | tr -d '.') && \
+    pip install --no-cache-dir torch==${PYTORCH_VERSION} torchaudio==${PYTORCH_VERSION} torchvision==${TORCHVISION_VERSION} torchvision torchaudio \
+    --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION_SANITIZED}
+
+# 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