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:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user