2 files changed, 23 insertions, 0 deletions
diff --git a/ci/ciimage/cuda/image.json b/ci/ciimage/cuda/image.json
new file mode 100644
index 000000000..d395e7197
--- /dev/null
+++ b/ci/ciimage/cuda/image.json
@@ -0,0 +1,5 @@
+{
+  "base_image": "archlinux:latest",
+  "args": ["--only", "cuda"],
+  "env": {}
+}
diff --git a/ci/ciimage/cuda/install.sh b/ci/ciimage/cuda/install.sh
new file mode 100755
index 000000000..011099d49
--- /dev/null
+++ b/ci/ciimage/cuda/install.sh
@@ -0,0 +1,18 @@
+#!/bin/bash
+
+set -e
+
+pkgs=(
+  python python-setuptools python-wheel python-pytest-xdist python-jsonschema
+  ninja gcc gcc-objc git cmake
+  cuda zlib pkgconf
+)
+
+PACMAN_OPTS='--needed --noprogressbar --noconfirm'
+
+pacman -Syu $PACMAN_OPTS "${pkgs[@]}"
+
+# Manually remove cache to avoid GitHub space restrictions
+rm -rf /var/cache/pacman
+
+echo "source /etc/profile.d/cuda.sh" >> /ci/env_vars.sh