fix(setup.py): Broaden the supported pytorch versions to handle jetson

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Feb 20, 2021 · e94a040 · e94a040
1 parent 1c9dfe2
commit e94a040
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py/setup.py b/py/setup.py
@@ -204,7 +204,7 @@ def run(self):
       long_description=long_description,
       ext_modules=ext_modules,
       install_requires=[
-          'torch==1.7.1',
+          'torch>=1.7.0,<1.8.0',
       ],
       setup_requires=[],
       cmdclass={