diff --git a/requirements.txt b/requirements.txt index 2b92917..554b013 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ # -*- coding: utf-8 -*- numpy -torch>=1.1 -torchvision>=0.4 +torch==2.3.1 +torchvision==0.18.1 scikit-learn yacs tqdm diff --git a/setup.py b/setup.py index 5adbb27..b4a15a9 100644 --- a/setup.py +++ b/setup.py @@ -192,7 +192,7 @@ if __name__ == '__main__': ], license='Apache License 2.0', install_requires=[ - 'numpy', 'torch>=1.2', 'torchvision>=0.4', 'scikit-learn', 'yacs', 'tqdm', + 'numpy', 'torch==2.3.1', 'torchvision==0.18.1', 'scikit-learn', 'yacs', 'tqdm', ], packages=find_packages(), zip_safe=False)