diff --git a/base/Dockerfile b/base/Dockerfile
index 69c035230..9785487c9 100644
--- a/base/Dockerfile
+++ b/base/Dockerfile
@@ -1,5 +1,5 @@
 # Ubuntu 24.04
-FROM nvcr.io/nvidia/cuda:12.9.1-base-ubuntu24.04 AS noble
+FROM nvcr.io/nvidia/cuda:13.1.0-base-ubuntu24.04 AS noble
 
 SHELL ["/bin/bash", "-c"]
 
@@ -36,7 +36,7 @@ RUN chmod +x /usr/local/bin/generate-ci-config && \
 ENTRYPOINT ["/usr/bin/sleep","1000"]
 
 # Ubuntu 22.04
-FROM nvcr.io/nvidia/cuda:12.9.1-base-ubuntu22.04 AS jammy
+FROM nvcr.io/nvidia/cuda:13.1.0-base-ubuntu22.04 AS jammy
 
 SHELL ["/bin/bash", "-c"]