diff --git a/pyproject.toml b/pyproject.toml index ba22227..bfbc34e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -34,7 +34,7 @@ dependencies = [ "pydmd >0.4, <= 0.4.1", "optht ~= 0.2.0", "prettytable >3.0.0, <= 3.9.0", - "torch ~= 2.1.0", + "torch >= 2.1,< 2.3", "torchvision ~= 0.16.0", "torchaudio ~= 2.1.0", "lightning ~= 2.0.9", diff --git a/requirements-torch.txt b/requirements-torch.txt index 113c633..8b9f868 100644 --- a/requirements-torch.txt +++ b/requirements-torch.txt @@ -1,3 +1,3 @@ -torch == 2.1.0+cu121 +torch == 2.2.0 torchvision lightning