diff --git a/base/Dockerfile b/base/Dockerfile
index d4fbd1cb..698923f6 100644
--- a/base/Dockerfile
+++ b/base/Dockerfile
@@ -1,5 +1,5 @@
 # Ubuntu 22.04
-FROM nvcr.io/nvidia/cuda:12.5.1-base-ubuntu22.04 as jammy
+FROM nvcr.io/nvidia/cuda:12.6.1-base-ubuntu22.04 as jammy
 
 SHELL ["/bin/bash", "-c"]
 
@@ -34,7 +34,7 @@ RUN chmod +x /usr/local/bin/generate-ci-config && \
 ENTRYPOINT ["/usr/bin/sleep","1000"]
 
 # Ubuntu 20.04
-FROM nvcr.io/nvidia/cuda:12.5.1-base-ubuntu20.04 as focal
+FROM nvcr.io/nvidia/cuda:12.6.1-base-ubuntu20.04 as focal
 
 SHELL ["/bin/bash", "-c"]