From bd6f20d989a79c34438b37aff9b28be4eb77aa57 Mon Sep 17 00:00:00 2001
From: Christian Legnitto <christian@legnitto.com>
Date: Wed, 1 Oct 2025 00:09:01 -0700
Subject: [PATCH] Add modal CI CUDA image

---
 .github/workflows/container_images.yml | 5 +++++
 container/modal-ci/Dockerfile          | 8 ++++++++
 2 files changed, 13 insertions(+)
 create mode 100644 container/modal-ci/Dockerfile

diff --git a/.github/workflows/container_images.yml b/.github/workflows/container_images.yml
index c056d144..29ac6568 100644
--- a/.github/workflows/container_images.yml
+++ b/.github/workflows/container_images.yml
@@ -42,6 +42,9 @@ jobs:
           - name: RockyLinux-9/CUDA-12.8.1
             image: "rust-gpu/rust-cuda-rockylinux9-cuda12"
             dockerfile: ./container/rockylinux9-cuda12/Dockerfile
+          - name: Ubuntu-24.04/CUDA-12.8.1/Modal-CI
+            image: "rust-gpu/rust-cuda-modal-ci"
+            dockerfile: ./container/modal-ci/Dockerfile
     steps:
       - name: Checkout repository
         uses: actions/checkout@v4
@@ -124,6 +127,8 @@ jobs:
             image: "rust-gpu/rust-cuda-ubuntu24-cuda12"
           - name: RockyLinux-9/CUDA-12.8.1
             image: "rust-gpu/rust-cuda-rockylinux9-cuda12"
+          - name: Ubuntu-24.04/CUDA-12.8.1/Modal-CI
+            image: "rust-gpu/rust-cuda-modal-ci"
     steps:
       - name: Set artifact name
         run: |
diff --git a/container/modal-ci/Dockerfile b/container/modal-ci/Dockerfile
new file mode 100644
index 00000000..68c0d440
--- /dev/null
+++ b/container/modal-ci/Dockerfile
@@ -0,0 +1,8 @@
+FROM ghcr.io/rust-gpu/rust-cuda-ubuntu24-cuda12:latest
+
+RUN apt-get update && \
+    DEBIAN_FRONTEND=noninteractive apt-get install -y python3 python3-pip python3-venv && \
+    ln -sf /usr/bin/python3 /usr/local/bin/python && \
+    apt-get clean && rm -rf /var/lib/apt/lists/*
+
+CMD ["python", "--version"]