feat: Update Pytorch version to 1.12

peri044 · peri044 · commit a18bb7ca3504 · 2022-07-11T19:02:10.000-07:00
Signed-off-by: Dheeraj Peri &lt;peri.dheeraj@gmail.com&gt;
diff --git a/README.md b/README.md
@@ -111,7 +111,7 @@ torch.jit.save(trt_ts_module, "trt_torchscript_module.ts") # save the TRT embedd
 These are the following dependencies used to verify the testcases. Torch-TensorRT can work with other versions, but the tests are not guaranteed to pass.
 
 - Bazel 5.1.1
-- Libtorch 1.11.0 (built with CUDA 11.3)
+- Libtorch 1.12.0 (built with CUDA 11.3)
 - CUDA 11.3
 - cuDNN 8.2.1
 - TensorRT 8.2.4.2
diff --git a/WORKSPACE b/WORKSPACE
@@ -56,17 +56,17 @@ new_local_repository(
 http_archive(
     name = "libtorch",
     build_file = "@//third_party/libtorch:BUILD",
-    sha256 = "8d9e829ce9478db4f35bdb7943308cf02e8a2f58cf9bb10f742462c1d57bf287",
+    sha256 = "80f089939de20e68e3fcad4dfa72a26c8bf91b5e77b11042f671f39ebac35865",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.11.0%2Bcu113.zip"],
+    urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.12.0%2Bcu113.zip"],
 )
 
 http_archive(
     name = "libtorch_pre_cxx11_abi",
     build_file = "@//third_party/libtorch:BUILD",
-    sha256 = "90159ecce3ff451f3ef3f657493b6c7c96759c3b74bbd70c1695f2ea2f81e1ad",
+    sha256 = "8e35371403f7052d9e9b43bcff383980dbde4df028986dc1dab539953481d55f",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.11.0%2Bcu113.zip"],
+    urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.12.0%2Bcu113.zip"],
 )
 
 # Download these tarballs manually from the NVIDIA website
diff --git a/py/requirements.txt b/py/requirements.txt
@@ -1,5 +1,5 @@
 -f https://download.pytorch.org/whl/torch_stable.html
 -f https://download.pytorch.org/whl/torch/
 --extra-index-url https://download.pytorch.org/whl/cu113
-torch==1.11.0+cu113
+torch==1.12.0+cu113
 pybind11==2.6.2
diff --git a/tests/modules/requirements.txt b/tests/modules/requirements.txt
@@ -1,4 +1,3 @@
 -f https://download.pytorch.org/whl/torch_stable.html
-#torch==1.11.0+cu113
 timm==v0.4.12
 transformers==4.17.0
diff --git a/tests/py/requirements.txt b/tests/py/requirements.txt
@@ -1,2 +1,2 @@
-torchvision==0.12.0+cu113
+torchvision==0.13.0+cu113
 -f https://download.pytorch.org/whl/torch_stable.html

Original file line number	Diff line number	Diff line change
`@@ -1,2 +1,2 @@`
`1`		`-torchvision==0.12.0+cu113`
	`1`	`+torchvision==0.13.0+cu113`
`2`	`2`	`-f https://download.pytorch.org/whl/torch_stable.html`