Merge pull request #355 from NVIDIA/clean_up_python_version_requirements

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

Author: narendasan

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={