diff --git a/Dockerfile.cuda b/Dockerfile.cuda
index 7a29b5e..b7900d1 100644
--- a/Dockerfile.cuda
+++ b/Dockerfile.cuda
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.6.1-cudnn-runtime-ubuntu22.04
+FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
 
 ARG FACEFUSION_VERSION=3.0.0
 ENV GRADIO_SERVER_NAME=0.0.0.0
diff --git a/Dockerfile.tensorrt b/Dockerfile.tensorrt
index 4081a4c..d069b7c 100644
--- a/Dockerfile.tensorrt
+++ b/Dockerfile.tensorrt
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.6.1-cudnn-runtime-ubuntu22.04
+FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
 
 ARG FACEFUSION_VERSION=3.0.0
 ENV GRADIO_SERVER_NAME=0.0.0.0