feat: sync pytorch for docker with version used in prepare_environment

This commit is contained in:
Manuel Schmid
2024-05-22 23:50:27 +02:00
parent 299ba4d2a2
commit 7f6dfe25a5
2 changed files with 3 additions and 6 deletions
+1 -1
View File
@@ -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