From 974e2398a57b4140e4fdc16f87f9dfd3469d1020 Mon Sep 17 00:00:00 2001
From: David bennett <david.a.bennett7@gmail.com>
Date: Sat, 14 Mar 2026 20:24:15 -0400
Subject: [PATCH] feat: add CUDA Dockerfile for GPU-accelerated builds

Multi-stage build using nvidia/cuda base images with
parameterized CUDA and Ubuntu versions. Follows the same
pattern as Dockerfile.vulkan and Dockerfile.musa.
---
 Dockerfile.cuda | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 Dockerfile.cuda

diff --git a/Dockerfile.cuda b/Dockerfile.cuda
new file mode 100644
index 000000000..b418a7ffe
--- /dev/null
+++ b/Dockerfile.cuda
@@ -0,0 +1,29 @@
+ARG UBUNTU_VERSION=22.04
+# This needs to generally match the container host's environment.
+ARG CUDA_VERSION=12.4.0
+# Target the CUDA build image
+ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
+
+ARG BASE_CUDA_RUN_CONTAINER=nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}
+
+FROM ${BASE_CUDA_DEV_CONTAINER} AS build
+
+RUN apt-get update && apt-get install -y --no-install-recommends build-essential git cmake
+
+WORKDIR /sd.cpp
+
+COPY . .
+
+RUN cmake . -B ./build -DSD_CUDA=ON
+RUN cmake --build ./build --config Release --parallel
+
+FROM ${BASE_CUDA_RUN_CONTAINER} AS runtime
+
+RUN apt-get update && \
+    apt-get install --yes --no-install-recommends libgomp1 && \
+    apt-get clean
+
+COPY --from=build /sd.cpp/build/bin/sd-cli /sd-cli
+COPY --from=build /sd.cpp/build/bin/sd-server /sd-server
+
+ENTRYPOINT [ "/sd-cli" ]