diff --git a/docker/cuda_simple/Dockerfile b/docker/cuda_simple/Dockerfile
index 0bbf20ffe..05e20d465 100644
--- a/docker/cuda_simple/Dockerfile
+++ b/docker/cuda_simple/Dockerfile
@@ -1,4 +1,4 @@
-ARG CUDA_IMAGE="12.5.0-devel-ubuntu22.04"
+ARG CUDA_IMAGE="12.6.1-devel-ubuntu24.04"
 FROM nvidia/cuda:${CUDA_IMAGE}
 
 # We need to set the host to 0.0.0.0 to allow outside access