diff --git a/docker.md b/docker.md
index 6dd58c20..0e1bc2a1 100644
--- a/docker.md
+++ b/docker.md
@@ -1,6 +1,6 @@
 # Fooocus on Docker
 
-The docker image is based on NVIDIA CUDA 12.4 and PyTorch 2.0, see [Dockerfile](Dockerfile) and [requirements_docker.txt](requirements_docker.txt) for details.
+The docker image is based on NVIDIA CUDA 12.4 and PyTorch 2.1, see [Dockerfile](Dockerfile) and [requirements_docker.txt](requirements_docker.txt) for details.
 
 ## Requirements
 
diff --git a/requirements_docker.txt b/requirements_docker.txt
index 3cf4aa89..21883adf 100644
--- a/requirements_docker.txt
+++ b/requirements_docker.txt
@@ -1,5 +1,2 @@
-torch==2.0.1
-torchvision==0.15.2
-torchaudio==2.0.2
-torchtext==0.15.2
-torchdata==0.6.1
+torch==2.1.0
+torchvision==0.16.0