Update CUDA base image version in Dockerfile
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user