From 7f6dfe25a5b1312e6ed41e7e6866344505971c69 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Wed, 22 May 2024 23:50:27 +0200 Subject: [PATCH] feat: sync pytorch for docker with version used in prepare_environment --- docker.md | 2 +- requirements_docker.txt | 7 ++----- 2 files changed, 3 insertions(+), 6 deletions(-) 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