diff --git a/rhel9/Dockerfile b/rhel9/Dockerfile index 98c48e5f..64780362 100644 --- a/rhel9/Dockerfile +++ b/rhel9/Dockerfile @@ -1,4 +1,4 @@ -FROM nvcr.io/nvidia/cuda:12.5.0-base-ubi9 as build +FROM nvcr.io/nvidia/cuda:12.5.1-base-ubi9 as build ARG TARGETARCH @@ -22,7 +22,7 @@ RUN git clone https://github.com/NVIDIA/gpu-driver-container driver && \ go build -o vgpu-util && \ mv vgpu-util /work -FROM nvcr.io/nvidia/cuda:12.5.0-base-ubi9 +FROM nvcr.io/nvidia/cuda:12.5.1-base-ubi9 ARG TARGETARCH ENV TARGETARCH=$TARGETARCH