diff --git a/ubuntu20.04/Dockerfile b/ubuntu20.04/Dockerfile
index 0ee623b0..9d6e3517 100644
--- a/ubuntu20.04/Dockerfile
+++ b/ubuntu20.04/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/cuda:12.6.0-base-ubuntu20.04 as build
+FROM nvcr.io/nvidia/cuda:12.6.1-base-ubuntu20.04 as build
 
 ARG TARGETARCH
 
@@ -33,7 +33,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.6.0-base-ubuntu20.04
+FROM nvcr.io/nvidia/cuda:12.6.1-base-ubuntu20.04
 
 SHELL ["/bin/bash", "-c"]