diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 552cbdd..21982a0 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -1,4 +1,4 @@
-FROM rocm/dev-ubuntu-24.04:6.4.4-complete
+FROM rocm/dev-ubuntu-24.04:6.4.4
 
 ARG FACEFUSION_VERSION=3.4.1
 ENV GRADIO_SERVER_NAME=0.0.0.0
@@ -8,6 +8,11 @@ WORKDIR /facefusion
 
 RUN apt-get update
 
+RUN apt-get install rocblas -y
+RUN apt-get install hipblas -y
+RUN apt-get install miopen-hip -y
+RUN apt-get install hipfft -y
+
 RUN apt-get install python3.12 -y
 RUN apt-get install python-is-python3 -y
 RUN apt-get install pip -y