Update CUDA base image version in Dockerfile

This commit is contained in:
McCloudS
2025-12-13 12:09:10 -05:00
committed by GitHub
parent 771ec16597
commit 512a8a473b

View File

@@ -22,7 +22,7 @@ RUN pip install --no-cache-dir -r requirements.txt
COPY . .
# Stage 2: Runtime
FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04
FROM nvidia/cuda:12.6.3-base-ubuntu22.04
WORKDIR /subgen