From 96b6c930e9377bbb386438cb6c7a317ca4cf85a1 Mon Sep 17 00:00:00 2001 From: Weilin Xu Date: Wed, 12 Jul 2023 16:02:39 -0700 Subject: [PATCH 1/2] Allow PyTorch 2.0.0 --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 6a80d751..5b200586 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -11,7 +11,7 @@ authors = [ requires-python = ">=3.9" dependencies = [ - "torch ~= 2.0.1", + "torch ~= 2.0.0", # Allow 2.0.0, because Poetry gets incomplete dependency for 2.0.1: https://github.com/python-poetry/poetry/issues/7902 "torchvision ~= 0.15.2", "lightning[extra] ~= 2.0.5", # Full functionality including TensorboardX. "pydantic == 1.10.11", # https://github.com/Lightning-AI/lightning/pull/18022/files From 9dfa18a8bfc5c16ec907f4cd699f3391dac43836 Mon Sep 17 00:00:00 2001 From: Weilin Xu Date: Wed, 12 Jul 2023 16:12:42 -0700 Subject: [PATCH 2/2] Also allow torchvision for torch 2.0.0. --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 5b200586..9015ac05 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -12,7 +12,7 @@ requires-python = ">=3.9" dependencies = [ "torch ~= 2.0.0", # Allow 2.0.0, because Poetry gets incomplete dependency for 2.0.1: https://github.com/python-poetry/poetry/issues/7902 - "torchvision ~= 0.15.2", + "torchvision ~= 0.15", "lightning[extra] ~= 2.0.5", # Full functionality including TensorboardX. "pydantic == 1.10.11", # https://github.com/Lightning-AI/lightning/pull/18022/files "torchmetrics == 1.0.0",