From 217673208571bceac73405beb896723c7c4262fa Mon Sep 17 00:00:00 2001
From: niuyazhe <niuyazhe@sensetime.com>
Date: Thu, 30 Sep 2021 14:56:27 +0800
Subject: [PATCH] fix(nyz): disable pytorch1.9.0 in default setting

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 47beb35..53abfec 100755
--- a/setup.py
+++ b/setup.py
@@ -48,7 +48,7 @@ setup(
         'requests>=2.25.1',
         'six',
         'gym>=0.20.0',  # pypy incompatible
-        'torch>=1.3.1,<=1.9.0',
+        'torch>=1.3.1,<=1.8.0',  # PyTorch 1.9.0 is available, but you need to do something like https://github.com/opendilab/DI-engine/discussions/81
         'pyyaml',
         'easydict==1.9',
         'tensorboardX>=2.1,<=2.2',
-- 
GitLab