diff --git a/Dockerfile.cpu b/Dockerfile.cpu
index 8d0cd83..a691df3 100644
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -1,6 +1,6 @@
 FROM python:3.10
 
-ARG FACEFUSION_VERSION=2.0.0
+ARG FACEFUSION_VERSION=2.1.0
 ENV GRADIO_SERVER_NAME=0.0.0.0
 
 WORKDIR /facefusion
@@ -10,4 +10,4 @@ RUN apt-get install curl -y
 RUN apt-get install ffmpeg -y
 
 RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
-RUN python install.py --torch cpu --onnxruntime default
+RUN python install.py --torch cpu --onnxruntime default --skip-venv
diff --git a/Dockerfile.cuda b/Dockerfile.cuda
index f5dd630..c5d01df 100644
--- a/Dockerfile.cuda
+++ b/Dockerfile.cuda
@@ -1,6 +1,6 @@
 FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
-ARG FACEFUSION_VERSION=2.0.0
+ARG FACEFUSION_VERSION=2.1.0
 ENV GRADIO_SERVER_NAME=0.0.0.0
 
 WORKDIR /facefusion
@@ -14,5 +14,6 @@ RUN apt-get install curl -y
 RUN apt-get install ffmpeg -y
 
 RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
-RUN python install.py --torch cuda --onnxruntime cuda
+RUN python install.py --torch cuda --onnxruntime cuda --skip-venv
+
 RUN cd /usr/local/lib/python3.10/dist-packages/torch/lib && ln -s libnvrtc-672ee683.so.11.2 libnvrtc.so
diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 5376949..9ee6814 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -1,35 +1,17 @@
-FROM rocm/dev-ubuntu-22.04:5.4.2-complete
+FROM rocm/dev-ubuntu-22.04:5.7.1
 
-ARG FACEFUSION_VERSION=2.0.0
-ARG ONNXRUNTIME_VERSION=1.15.0
-ARG ROCM_VERSION=5.4.2
-ARG CMAKE_VERSION=3.27.4
+ARG FACEFUSION_VERSION=2.1.0
 ENV GRADIO_SERVER_NAME=0.0.0.0
 
+WORKDIR /facefusion
+
 RUN apt-get update
-RUN apt-get install python3-dev -y
+RUN apt-get install python3.10 -y
+RUN apt-get install python-is-python3 -y
+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 numpy
-RUN pip install flake8
-RUN pip install packaging
-RUN pip install wheel
-
-RUN curl -L https://github.com/Kitware/CMake/releases/download/v${CMAKE_VERSION}/cmake-${CMAKE_VERSION}-linux-x86_64.tar.gz | tar xz -C /opt
-RUN sudo ln -sf /opt/cmake-${CMAKE_VERSION}-linux-x86_64/bin/cmake /usr/bin/cmake
-RUN sudo ln -sf /opt/cmake-${CMAKE_VERSION}-linux-x86_64/bin/ctest /usr/bin/ctest
-
-WORKDIR /onnxruntime
-
-RUN git clone https://github.com/Microsoft/onnxruntime --branch v${ONNXRUNTIME_VERSION} --single-branch .
-RUN chmod +x /onnxruntime/tools/ci_build/build.py
-RUN /onnxruntime/tools/ci_build/build.py --allow_running_as_root --build_dir build --config Release --build_wheel --update --build --parallel --skip_tests --cmake_extra_defines ONNXRUNTIME_VERSION=${ONNXRUNTIME_VERSION} --use_rocm --rocm_home /opt/rocm
-
-WORKDIR /facefusion
-
 RUN git clone https://github.com/facefusion/facefusion.git --branch ${FACEFUSION_VERSION} --single-branch .
-RUN pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/rocm${ROCM_VERSION}
-RUN pip uninstall onnxruntime -y
-RUN pip install /onnxruntime/build/Release/dist/*.whl
+RUN python install.py --torch rocm --onnxruntime rocm --skip-venv