diff --git a/Dockerfile b/Dockerfile
index da74f31..caaa86b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 # Use the specified CUDA base image with Rocky Linux 9
-FROM nvidia/cuda:12.8.1-cudnn-devel-rockylinux9
+FROM nvidia/cuda:12.9.1-cudnn-devel-rockylinux9
 
 # Install dependencies
 WORKDIR /app