diff --git a/setup.py b/setup.py
index 47beb35ced707ac5a2a62fd6603997adcb3eb29c..53abfecfe5156af571d6186295f246f515e8a6b7 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',