diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index 43f1abd..f99bd7a 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -8,7 +8,7 @@
 # https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html and
 # adjust the base image tag to match. This Dockerfile has not been build/run against
 # real GPU hardware in this environment - verify on your own target before relying on it.
-FROM nvidia/cuda:13.3.1-cudnn-runtime-ubuntu22.04
+FROM nvidia/cuda:13.4.1-cudnn-runtime-ubuntu22.04
 
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \