diff --git a/py/setup.py b/py/setup.py index ceaa681016..a9913084ec 100644 --- a/py/setup.py +++ b/py/setup.py @@ -299,7 +299,7 @@ def run(self): long_description=long_description, ext_modules=ext_modules, install_requires=[ - 'torch>=1.11.0+cu113<1.12.0', + 'torch>=1.11.0+cu113,<1.12.0', ], setup_requires=[], cmdclass={