Merge pull request #287 from ahmetoner/reduce-gpu-image-size

Update CUDA base image to reduce image size

CHANGELOG.md

@@ -4,6 +4,10 @@ Changelog
 Unreleased
 ----------
 
+### Changed
+
+- Reduced GPU image size by using `nvidia/cuda:12.6.3-base-ubuntu22.04`
+
 [1.8.1] (2025-02-18)
 --------------------
 

Dockerfile.gpu

@@ -2,7 +2,7 @@ FROM onerahmet/ffmpeg:n7.1 AS ffmpeg
 
 FROM swaggerapi/swagger-ui:v5.9.1 AS swagger-ui
 
-FROM nvidia/cuda:12.6.3-cudnn-runtime-ubuntu22.04
+FROM nvidia/cuda:12.6.3-base-ubuntu22.04
 
 ENV PYTHON_VERSION=3.10