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]>

Author: narendasan

Diff for: 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={