happyz
/
Fooocus
mirror of https://github.com/lllyasviel/Fooocus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
2 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docker.md
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

7
requirements_docker.txt
View File

@ -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