diff --git a/demo/gpu-error/illegal-memory-access/Dockerfile b/demo/gpu-error/illegal-memory-access/Dockerfile
index 15ec924e3..6cc500f29 100644
--- a/demo/gpu-error/illegal-memory-access/Dockerfile
+++ b/demo/gpu-error/illegal-memory-access/Dockerfile
@@ -14,7 +14,7 @@
 
 # Use a recent CUDA development image (includes nvcc compiler)
 # --- Stage 1: Build ---
-FROM nvidia/cuda:13.0.2-devel-ubuntu22.04 AS builder
+FROM nvidia/cuda:13.1.0-devel-ubuntu22.04 AS builder
 
 WORKDIR /app
 COPY vectorAdd.cu .
@@ -24,7 +24,7 @@ RUN nvcc vectorAdd.cu -o vectorAdd
 
 # --- Stage 2: Runtime ---
 # Use a smaller base image containing only the CUDA runtime and nvidia-smi
-FROM nvidia/cuda:13.0.2-base-ubuntu22.04
+FROM nvidia/cuda:13.1.0-base-ubuntu22.04
 
 WORKDIR /app
 # Copy only the compiled executable from the build stage