Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: sync pytorch for docker with version used in prepare_environment

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
docker.md
+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
requirements_docker.txt
+2 -5
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