Pytorch 1.7.1 update

Signed-off-by: Dheeraj Peri <[email protected]>

Author: peri044

README.md

@@ -74,7 +74,7 @@ torch.jit.save(trt_ts_module, "trt_torchscript_module.ts")
 These are the following dependencies used to verify the testcases. TRTorch can work with other versions, but the tests are not guaranteed to pass.
 
 - Bazel 3.7.0
-- Libtorch 1.7.0 (built with CUDA 11.0)
+- Libtorch 1.7.1 (built with CUDA 11.0)
 - CUDA 11.0
 - cuDNN 8
 - TensorRT 7.2.1.6

WORKSPACE

@@ -53,16 +53,16 @@ http_archive(
     name = "libtorch",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    sha256 = "656db919c00e99dac81bc21598845ba9231e57cbc8e1570bea017dfb9298236d",
-    urls = ["https://download.pytorch.org/libtorch/cu110/libtorch-cxx11-abi-shared-with-deps-1.7.0%2Bcu110.zip"],
+    sha256 = "117f6dd65b7267839197397edd0b10fd2900b0f291e3e54b0b800caefc31bcb6",
+    urls = ["https://download.pytorch.org/libtorch/cu110/libtorch-cxx11-abi-shared-with-deps-1.7.1%2Bcu110.zip"],
 )
 
 http_archive(
     name = "libtorch_pre_cxx11_abi",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    sha256 = "b3976f3294a6c20be4807191042d1e74162a41fb1bed0919f637afc3517d76be",
-    urls = ["https://download.pytorch.org/libtorch/cu110/libtorch-shared-with-deps-1.7.0%2Bcu110.zip"],
+    sha256 = "c77f926afd55d7e860ec9c7abc992c25be77c89771c3ec6fcc13ea42f07d46df",
+    urls = ["https://download.pytorch.org/libtorch/cu110/libtorch-shared-with-deps-1.7.1%2Bcu110.zip"],
 )
 
 # Download these tarballs manually from the NVIDIA website
@@ -85,7 +85,6 @@ http_archive(
     strip_prefix = "TensorRT-7.2.1.6"
 )
 
-
 ####################################################################################
 # Locally installed dependencies (use in cases of custom dependencies or aarch64)
 ####################################################################################

py/setup.py

@@ -204,7 +204,7 @@ def run(self):
       long_description=long_description,
       ext_modules=ext_modules,
       install_requires=[
-          'torch==1.7.0',
+          'torch==1.7.1',
       ],
       setup_requires=[],
       cmdclass={

tests/modules/hub.py

@@ -46,15 +46,15 @@
         "path": "both"
     },
     "resnet18": {
-        "model": torch.hub.load('pytorch/vision:v0.8.0', 'resnet18', pretrained=True),
+        "model": torch.hub.load('pytorch/vision:v0.8.2', 'resnet18', pretrained=True),
         "path": "both"
     },
     "resnet50": {
-        "model": torch.hub.load('pytorch/vision:v0.8.0', 'resnet50', pretrained=True),
+        "model": torch.hub.load('pytorch/vision:v0.8.2', 'resnet50', pretrained=True),
         "path": "both"
     },
     "fcn_resnet101": {
-        "model": torch.hub.load('pytorch/vision:v0.8.0', 'fcn_resnet101', pretrained=True),
+        "model": torch.hub.load('pytorch/vision:v0.8.2', 'fcn_resnet101', pretrained=True),
         "path": "script"
     },
     "ssd": {

tests/py/requirements.txt

@@ -1 +1 @@
-torchvision==0.8.0
+torchvision==0.8.2