diff --git a/docker/Dockerfile b/docker/Dockerfile
index 483c9989fe0a30feadfaddfee636fd698fa83a08..d54c85c60d74ce276506363dcca3c8535faebcc9 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,5 +1,5 @@
 # Base image with Ubuntu 22.04
-FROM nvidia/cuda:12.0-base-ubuntu22.04
+FROM nvidia/cuda:12.6.0-base-ubuntu22.04
 
 # Install dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \