From 34466fcd2273136a96b60749f46672079637537a Mon Sep 17 00:00:00 2001
From: Gitanshu Sardana <gitanshusardana@gmail.com>
Date: Mon, 7 Jun 2021 17:56:07 -0700
Subject: [PATCH] Unpin torch and torchvision (ref #2248)

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 0a1e0ff317..a2e04f0740 100644
--- a/setup.py
+++ b/setup.py
@@ -24,8 +24,8 @@
     'setproctitle>=1.0',
     'tensorflow>=1.14',
     'tensorflow-probability>=0.11.0',
-    'torch>=1.0.0,!=1.5.0,<1.8.0',
-    'torchvision>=0.2.1,<=0.8.2',
+    'torch>=1.0.0,!=1.5.0,!=1.8.0',
+    'torchvision>=0.2.1',
 ]
 
 if sys.version_info < (3, 7):