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

ahmetoner · web-flow · commit 338faafedcc9 · 2025-02-18T01:52:01.000+01:00
Update CUDA base image to reduce image size
diff --git a/CHANGELOG.md b/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)
 --------------------
 
diff --git a/Dockerfile.gpu b/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
 
