From d3b9408e70c29b6092cdf1b4f4d26bc8c8dc7835 Mon Sep 17 00:00:00 2001
From: Gabby <gromano@lightricks.com>
Date: Wed, 18 Feb 2026 00:02:25 +0200
Subject: [PATCH] feat: add cuda-nvcc to runner image for CUDA kernel
 compilation

Adds cuda-nvcc-12-2 and cuda-cudart-dev-12-2 packages to enable
compilation of CUDA kernels (e.g., q8_kernels_ltx) on H100 runners.

Also sets PATH and LD_LIBRARY_PATH for CUDA 12.2.
---
 images/Dockerfile | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/images/Dockerfile b/images/Dockerfile
index 7eabd13..3c88396 100644
--- a/images/Dockerfile
+++ b/images/Dockerfile
@@ -6,11 +6,15 @@ FROM nvidia/cuda:12.2.0-base-ubuntu22.04 as base
 # Disable buffering for Python output
 ENV PYTHONUNBUFFERED=1
 ENV DEBIAN_FRONTEND=noninteractive
+ENV PATH=/usr/local/cuda-12.2/bin:${PATH}
+ENV LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:${LD_LIBRARY_PATH}
 
 # Install system dependencies, including Python 3.10
 RUN apt-get update && apt-get install -y --no-install-recommends software-properties-common \
     && add-apt-repository -y ppa:deadsnakes/ppa \
     && apt-get update && apt-get install --allow-downgrades --allow-change-held-packages --no-install-recommends -y \
+    cuda-nvcc-12-2 \
+    cuda-cudart-dev-12-2 \
     build-essential \
     cmake \
     curl \