diff --git a/docker/diffusers-pytorch-minimum-cuda/Dockerfile b/docker/diffusers-pytorch-minimum-cuda/Dockerfile index 00d077c5da60..20e10509da33 100644 --- a/docker/diffusers-pytorch-minimum-cuda/Dockerfile +++ b/docker/diffusers-pytorch-minimum-cuda/Dockerfile @@ -4,9 +4,9 @@ LABEL repository="diffusers" ARG PYTHON_VERSION=3.10 ENV DEBIAN_FRONTEND=noninteractive -ENV MINIMUM_SUPPORTED_TORCH_VERSION="2.1.0" -ENV MINIMUM_SUPPORTED_TORCHVISION_VERSION="0.16.0" -ENV MINIMUM_SUPPORTED_TORCHAUDIO_VERSION="2.1.0" +ENV MINIMUM_SUPPORTED_TORCH_VERSION="2.6.0" +ENV MINIMUM_SUPPORTED_TORCHVISION_VERSION="0.21.0" +ENV MINIMUM_SUPPORTED_TORCHAUDIO_VERSION="2.6.0" RUN apt-get -y update \ && apt-get install -y software-properties-common \ diff --git a/docs/source/en/installation.md b/docs/source/en/installation.md index abde3251de27..f56932463169 100644 --- a/docs/source/en/installation.md +++ b/docs/source/en/installation.md @@ -12,7 +12,7 @@ specific language governing permissions and limitations under the License. # Installation -Diffusers is tested on Python 3.8+ and PyTorch 1.4+. Install [PyTorch](https://pytorch.org/get-started/locally/) according to your system and setup. +Diffusers is tested on Python 3.8+ and PyTorch 2.6+. Install [PyTorch](https://pytorch.org/get-started/locally/) according to your system and setup. Create a [virtual environment](https://packaging.python.org/guides/installing-using-pip-and-virtual-environments/) for easier management of separate projects and to avoid compatibility issues between dependencies. Use [uv](https://docs.astral.sh/uv/), a Rust-based Python package and project manager, to create a virtual environment and install Diffusers. diff --git a/setup.py b/setup.py index 16d6b39aedf0..bc8110bbc594 100644 --- a/setup.py +++ b/setup.py @@ -137,7 +137,7 @@ "requests", "tensorboard", "tiktoken>=0.7.0", - "torch>=1.4", + "torch>=2.6", "torchvision", "transformers>=4.41.2", "urllib3<=2.0.0", diff --git a/src/diffusers/dependency_versions_table.py b/src/diffusers/dependency_versions_table.py index b8c337c2ad2e..747d1011aa40 100644 --- a/src/diffusers/dependency_versions_table.py +++ b/src/diffusers/dependency_versions_table.py @@ -44,7 +44,7 @@ "requests": "requests", "tensorboard": "tensorboard", "tiktoken": "tiktoken>=0.7.0", - "torch": "torch>=1.4", + "torch": "torch>=2.6", "torchvision": "torchvision", "transformers": "transformers>=4.41.2", "urllib3": "urllib3<=2.0.0",