Pin torchvision version until #2248 is fixed (#2252)

rlworkgroup · Mar 16, 2021 · 2abb519 · 2abb519
1 parent 6d53e49
commit 2abb519
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -25,7 +25,7 @@
     'tensorflow>=1.14',
     'tensorflow-probability>=0.11.0',
     'torch>=1.0.0,!=1.5.0,<1.8.0',
-    'torchvision>=0.2.1',
+    'torchvision>=0.2.1,<=0.8.2',
 ]
 
 if sys.version_info < (3, 7):