diff --git a/setup.py b/setup.py index 054fc3be6d..5a972b471d 100755 --- a/setup.py +++ b/setup.py @@ -29,7 +29,7 @@ setup( keywords=['deep learning', 'pytorch', 'AI'], python_requires='>=3.6', install_requires=[ - 'torch>=1.1.0', + 'torch==1.1.0', 'tqdm', 'test-tube>=0.6.7.6', 'pandas>=0.20.3',