diff --git a/requirements.dev.txt b/requirements.dev.txt index 7c8e5047..6c78701d 100644 --- a/requirements.dev.txt +++ b/requirements.dev.txt @@ -49,7 +49,7 @@ streamlit>=0.69.1 stringcase>=1.2.0 tabulate>=0.8.5 termgraph==0.5.3 -torch==1.6.0 +torch==1.13.1 torchsummary==1.5.1 tornado>=6.1 tqdm>=4.51.0 diff --git a/setup.py b/setup.py index 213cd09b..efc1e7db 100644 --- a/setup.py +++ b/setup.py @@ -147,7 +147,7 @@ setup( "requests>=2.25.1", "scipy>=1.7.0", "tabulate>=0.8.5", - "torch>=1.6.0, <1.8.0", + "torch>=1.6.0, <1.14.0", "tornado>=6.1", ] + specific_requires,