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 . .
|
COPY . .
|
||||||
|
|
||||||
# Stage 2: Runtime
|
# Stage 2: Runtime
|
||||||
FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04
|
FROM nvidia/cuda:12.6.3-base-ubuntu22.04
|
||||||
|
|
||||||
WORKDIR /subgen
|
WORKDIR /subgen
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user