Dockerfile.base

@@ -1,5 +1,5 @@
 # Base image with complete ML and hardware acceleration stack
-FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-devel
+FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-runtime
 
 # Install build dependencies and system libraries
 RUN apt-get update && apt-get install -y \
@@ -37,7 +37,13 @@ RUN apt-get update && apt-get install -y \
     python3-numpy \
     && rm -rf /var/lib/apt/lists/*
 
-# CUDA development tools already available in devel image
+# Install minimal CUDA development tools (just what we need for FFmpeg)
+RUN apt-get update && apt-get install -y \
+    cuda-nvcc-12-6 \
+    cuda-cudart-dev-12-6 \
+    libnvidia-encode-12-6 \
+    libnvidia-decode-12-6 \
+    && rm -rf /var/lib/apt/lists/*
 
 # Ensure CUDA paths are available
 ENV PATH="/usr/local/cuda/bin:${PATH}"