diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 87b297f..c4f2eaa 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -1,4 +1,4 @@
-FROM rocm/dev-ubuntu-24.04:6.2
+FROM rocm/dev-ubuntu-24.04:6.4
 
 ARG FACEFUSION_VERSION=3.2.0
 ENV GRADIO_SERVER_NAME=0.0.0.0
diff --git a/Dockerfile.tensorrt b/Dockerfile.tensorrt
index 569c589..13c79bb 100644
--- a/Dockerfile.tensorrt
+++ b/Dockerfile.tensorrt
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.8.0-cudnn-runtime-ubuntu24.04
+FROM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04
 
 ARG FACEFUSION_VERSION=3.2.0
 ENV GRADIO_SERVER_NAME=0.0.0.0
@@ -14,7 +14,7 @@ RUN apt-get install pip -y
 RUN apt-get install git -y
 RUN apt-get install curl -y
 RUN apt-get install ffmpeg -y
-RUN pip install tensorrt==10.8.0.43 --extra-index-url https://pypi.nvidia.com
+RUN pip install tensorrt==10.9.0.34 --extra-index-url https://pypi.nvidia.com
 
 RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
 RUN python install.py --onnxruntime cuda --skip-conda