diff --git a/Dockerfile b/Dockerfile
index b67dd5d..62ce5fa 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.1-cudnn9-runtime-ubuntu22.04
+FROM nvidia/cuda:12.1.0-cudnn9-runtime-ubuntu22.04
 
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive