Pin torch until #2248 is fixed (#2250)

rlworkgroup · Mar 13, 2021 · 6d53e49 · 6d53e49
1 parent 62bbc5c
commit 6d53e49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -24,7 +24,7 @@
     'setproctitle>=1.0',
     'tensorflow>=1.14',
     'tensorflow-probability>=0.11.0',
-    'torch>=1.0.0,!=1.5.0',
+    'torch>=1.0.0,!=1.5.0,<1.8.0',
     'torchvision>=0.2.1',
 ]