From fa3ab5c6d2a49e064258ca18f5963a0d7ecd011a Mon Sep 17 00:00:00 2001
From: Siwat Sirichai <siwat@siwatinc.com>
Date: Fri, 26 Sep 2025 00:48:39 +0700
Subject: [PATCH] refactor: update base image to runtime version and install
 minimal CUDA development tools for FFmpeg

---
 Dockerfile.base | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/Dockerfile.base b/Dockerfile.base
index 8d19778..2569ebd 100644
--- a/Dockerfile.base
+++ b/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}"