Downgrade CUDA base image version in Dockerfile

Not worth the other broken stuff to save ~1gb of space for now...
This commit is contained in:
McCloudS
2025-12-13 12:34:08 -05:00
committed by GitHub
parent 8bbf9eff6f
commit 6e99826c9e

View File

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