diff --git a/Dockerfile b/Dockerfile
index 704c3d71f..2aea28106 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,6 +10,7 @@ RUN apt-get update -y && \
 COPY requirements_docker.txt requirements_versions.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements_docker.txt -r /tmp/requirements_versions.txt && \
 	rm -f /tmp/requirements_docker.txt /tmp/requirements_versions.txt
+RUN pip install --no-cache-dir xformers==0.0.22 --no-dependencies
 RUN curl -fsL -o /usr/local/lib/python3.10/dist-packages/gradio/frpc_linux_amd64_v0.2 https://cdn-media.huggingface.co/frpc-gradio-0.2/frpc_linux_amd64 && \
 	chmod +x /usr/local/lib/python3.10/dist-packages/gradio/frpc_linux_amd64_v0.2
 
diff --git a/docker.md b/docker.md
index 9ae37c6fa..46fcb933a 100644
--- a/docker.md
+++ b/docker.md
@@ -1,6 +1,6 @@
 # Fooocus on Docker
 
-The docker image is based on NVIDIA CUDA 12.3 and PyTorch 2.1, see [Dockerfile](Dockerfile) and [requirements_docker.txt](requirements_docker.txt) for details.
+The docker image is based on NVIDIA CUDA 12.3 and PyTorch 2.0, see [Dockerfile](Dockerfile) and [requirements_docker.txt](requirements_docker.txt) for details.
 
 ## Quick start
 
diff --git a/requirements_docker.txt b/requirements_docker.txt
index de3d25360..3cf4aa89d 100644
--- a/requirements_docker.txt
+++ b/requirements_docker.txt
@@ -1,7 +1,5 @@
-torch==2.2.0
-torchvision==0.17.0
-torchaudio==2.2.0
-torchtext==0.17.0
-torchdata==0.7.1
-xformers==0.0.24
-triton==2.2.0
+torch==2.0.1
+torchvision==0.15.2
+torchaudio==2.0.2
+torchtext==0.15.2
+torchdata==0.6.1