chore!: Update dependencies to PyTorch 1.6.0

BREAKING CHANGE: Support for Python 3.5 is being dropped with this
update

Signed-off-by: Naren Dasan <naren@narendasan.com>
Signed-off-by: Naren Dasan <narens@nvidia.com>

.bazelversion

@@ -1 +1 @@
-3.3.1
+3.4.1

WORKSPACE

@@ -53,16 +53,16 @@ http_archive(
     name = "libtorch",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.5.1.zip"],
-    sha256 = "cf0691493d05062fe3239cf76773bae4c5124f4b039050dbdd291c652af3ab2a"
+    urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.6.0.zip"],
+    sha256 = "fded948bd2dbee625cee33ebbd4843a69496729389e0200a90fbb667cdaeeb69"
 )
 
 http_archive(
     name = "libtorch_pre_cxx11_abi",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    sha256 = "818977576572eadaf62c80434a25afe44dbaa32ebda3a0919e389dcbe74f8656",
-    urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-shared-with-deps-1.5.1.zip"],
+    sha256 = "141bb229f4bbf905541096cf8705785e7b0c79e37ca1e5db9d372730b1b9abd7",
+    urls = ["https://download.pytorch.org/libtorch/cu102/libtorch-shared-with-deps-1.6.0.zip"],
 )
 
 # Download these tarballs manually from the NVIDIA website

py/build_whl.sh

@@ -6,12 +6,6 @@ cd /workspace/TRTorch/py
 
 export CXX=g++
 
-build_py35() {
-    /opt/python/cp35-cp35m/bin/python -m pip install -r requirements.txt
-    /opt/python/cp35-cp35m/bin/python setup.py bdist_wheel
-    #auditwheel repair --plat manylinux2014_x86_64
-}
-
 build_py36() {
     /opt/python/cp36-cp36m/bin/python -m pip install -r requirements.txt
     /opt/python/cp36-cp36m/bin/python setup.py bdist_wheel
@@ -30,7 +24,6 @@ build_py38() {
     #auditwheel repair --plat manylinux2014_x86_64
 }
 
-build_py35
 build_py36
 build_py37
 build_py38

py/requirements.txt

@@ -1 +1 @@
-torch==1.5.1
+torch==1.6.0

py/setup.py

@@ -196,7 +196,7 @@ def run(self):
     long_description=long_description,
     ext_modules=ext_modules,
     install_requires=[
-        'torch==1.5.1',
+        'torch==1.6.0',
     ],
     setup_requires=[],
     cmdclass={
@@ -210,7 +210,7 @@ def run(self):
     license="BSD",
     packages=find_packages(),
     classifiers=[
-        "Development Status :: 3 - Alpha",
+        "Development Status :: 4 - Beta",
         "Environment :: GPU :: NVIDIA CUDA",
         "License :: OSI Approved :: BSD License",
         "Intended Audience :: Developers",
@@ -224,7 +224,7 @@ def run(self):
         "Topic :: Software Development",
         "Topic :: Software Development :: Libraries"
     ],
-    python_requires='>=3.5',
+    python_requires='>=3.6',
     include_package_data=True,
     package_data={
         'trtorch': ['lib/*.so'],

tests/py/requirements.txt

@@ -1 +1 @@
-torchvision==0.6.0
+torchvision==0.7.0