# Igneum GPU bench image, NVIDIA lane. CUDA 12.8 devel (nvcc, NVRTC, cuda.h) on Ubuntu 22.04, plus the OpenCL ICD
# loader and headers for the optional OpenCL build-time measurement on NVIDIA, Python 3 for run.sh, curl for the
# upload, and sshd so the bundle can be scp'd in. RunPod: create a template with this image (or use the tag below
# directly; nvcc is in it) and tick "SSH". Blackwell (RTX 5090) needs 12.8 or newer; older toolkits do not know sm_120.
# Build and push (optional; the public tag works as-is for RunPod with run.sh installing nothing):
#   docker build -f Dockerfile.cuda -t <you>/igneum-gpu-bench:cuda12.8 .
FROM nvidia/cuda:12.8.1-devel-ubuntu22.04
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y --no-install-recommends \
      python3 curl ca-certificates openssh-server clinfo ocl-icd-opencl-dev opencl-headers perl build-essential \
    && rm -rf /var/lib/apt/lists/* \
    && mkdir -p /run/sshd /workspace
# NVIDIA's OpenCL ICD is provided by the host driver through the container runtime; it appears when the pod sets
# NVIDIA_DRIVER_CAPABILITIES=compute,utility (RunPod does). clinfo shows whether it is there; run.sh skips if not.
WORKDIR /workspace
CMD ["/usr/sbin/sshd", "-D"]