diff --git a/setup.py b/setup.py
index 328d05d4e97bbaf767a9a973abd665124610e43d..03c533a39e4aab6fd5aadc1040642644c08f7995 100644
--- a/setup.py
+++ b/setup.py
@@ -10,8 +10,7 @@ REQUIREMENTS = [
     'requests==2.23.0',
     'overrides==3.0.0',
     'tensorboard==2.1.0',
-    'torch>=1.5.0',
-    'torchvision>=0.6.0',
+    'torch>=1.5.0,<1.6.0',
     'tqdm==4.43.0',
     'transformers==2.9.1',
     'urllib3==1.24.2',