Skip to content

Commit cc7d0b7

Browse files
peri044narendasan
peri044
authored and
narendasan
committed
refactor! : Update Pytorch version to 1.10
BREAKING CHANGE: This commit updates the pytorch version to 1.10. To use python API of torch_tensorrt, please upgrade your local pytorch to 1.10 to avoid ABI incompatibility errors. WORKSPACE and requirements files are updated accordingly Signed-off-by: Dheeraj Peri <[email protected]> Signed-off-by: Dheeraj Peri <[email protected]>
1 parent a8a407f commit cc7d0b7

File tree

9 files changed

+17
-18
lines changed

9 files changed

+17
-18
lines changed

Diff for: ‎README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ torch.jit.save(trt_ts_module, "trt_torchscript_module.ts")
9696
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.
9797

9898
- Bazel 4.0.0
99-
- Libtorch 1.9.1 (built with CUDA 11.1)
99+
- Libtorch 1.10.0 (built with CUDA 11.3)
100100
- CUDA 11.1 (10.2 on Jetson)
101101
- cuDNN 8.2
102102
- TensorRT 8.0.3.4 (TensorRT 8.0.1.6 on Jetson)

Diff for: ‎WORKSPACE

+4-4
Original file line numberDiff line numberDiff line change
@@ -56,17 +56,17 @@ new_local_repository(
5656
http_archive(
5757
name = "libtorch",
5858
build_file = "@//third_party/libtorch:BUILD",
59-
sha256 = "db57b1023fb33768286a98ba22c44cfe03d6ed158bc2dc0ca1d4928ee5f19f60",
59+
sha256 = "190e963e739d5f7c2dcf94b3994de8fcd335706a4ebb333812ea7d8c841beb06",
6060
strip_prefix = "libtorch",
61-
urls = ["https://download.pytorch.org/libtorch/cu111/libtorch-cxx11-abi-shared-with-deps-1.9.1%2Bcu111.zip"],
61+
urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.10.0%2Bcu113.zip"],
6262
)
6363

6464
http_archive(
6565
name = "libtorch_pre_cxx11_abi",
6666
build_file = "@//third_party/libtorch:BUILD",
67-
sha256 = "5563ca53b2b5342f1ab7eef9baf308f197673663ad5b1458a031c46dd802f413",
67+
sha256 = "0996a6a4ea8bbc1137b4fb0476eeca25b5efd8ed38955218dec1b73929090053",
6868
strip_prefix = "libtorch",
69-
urls = ["https://download.pytorch.org/libtorch/cu111/libtorch-shared-with-deps-1.9.1%2Bcu111.zip"],
69+
urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.10.0%2Bcu113.zip"],
7070
)
7171

7272
# Download these tarballs manually from the NVIDIA website

Diff for: ‎docker/WORKSPACE.docs

+4-5
Original file line numberDiff line numberDiff line change
@@ -52,17 +52,17 @@ new_local_repository(
5252
http_archive(
5353
name = "libtorch",
5454
build_file = "@//third_party/libtorch:BUILD",
55-
sha256 = "db57b1023fb33768286a98ba22c44cfe03d6ed158bc2dc0ca1d4928ee5f19f60",
55+
sha256 = "190e963e739d5f7c2dcf94b3994de8fcd335706a4ebb333812ea7d8c841beb06",
5656
strip_prefix = "libtorch",
57-
urls = ["https://download.pytorch.org/libtorch/cu111/libtorch-cxx11-abi-shared-with-deps-1.9.1%2Bcu111.zip"],
57+
urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.10.0%2Bcu113.zip"],
5858
)
5959

6060
http_archive(
6161
name = "libtorch_pre_cxx11_abi",
6262
build_file = "@//third_party/libtorch:BUILD",
63-
sha256 = "5563ca53b2b5342f1ab7eef9baf308f197673663ad5b1458a031c46dd802f413",
63+
sha256 = "0996a6a4ea8bbc1137b4fb0476eeca25b5efd8ed38955218dec1b73929090053",
6464
strip_prefix = "libtorch",
65-
urls = ["https://download.pytorch.org/libtorch/cu111/libtorch-shared-with-deps-1.9.1%2Bcu111.zip"],
65+
urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.10.0%2Bcu113.zip"],
6666
)
6767

6868
####################################################################################
@@ -99,4 +99,3 @@ pip3_import(
9999

100100
load("@py_test_deps//:requirements.bzl", "pip_install")
101101
pip_install()
102-

Diff for: ‎docsrc/tutorials/installation.rst

+1-1
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ the CUDA driver installed and the container must have CUDA)
6666

6767
The correct LibTorch version will be pulled down for you by bazel.
6868

69-
NOTE: For best compatability with official PyTorch, use torch==1.9.1+cuda111, TensorRT 8.0 and cuDNN 8.2 for CUDA 11.1 however Torch-TensorRT itself supports
69+
NOTE: For best compatability with official PyTorch, use torch==1.10.0+cuda113, TensorRT 8.0 and cuDNN 8.2 for CUDA 11.3 however Torch-TensorRT itself supports
7070
TensorRT and cuDNN for CUDA versions other than 11.1 for usecases such as using NVIDIA compiled distributions of PyTorch that use other versions of CUDA
7171
e.g. aarch64 or custom compiled version of PyTorch.
7272

Diff for: ‎examples/int8/training/vgg16/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
1-
torch>=1.9.1
1+
torch>=1.10.0
22
tensorboard>=1.14.0
33
pytorch-quantization --extra-index-url https://pypi.ngc.nvidia.com

Diff for: ‎py/requirements.txt

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
-f https://download.pytorch.org/whl/torch_stable.html
2-
torch==1.9.1+cu111
3-
pybind11==2.6.2
2+
torch==1.10.0+cu113
3+
pybind11==2.6.2

Diff for: ‎py/setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ def run(self):
249249
long_description=long_description,
250250
ext_modules=ext_modules,
251251
install_requires=[
252-
'torch>=1.9.0<1.11.0',
252+
'torch>=1.10.0<1.11.0',
253253
],
254254
setup_requires=[],
255255
cmdclass={

Diff for: ‎tests/modules/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
-f https://download.pytorch.org/whl/torch_stable.html
22
timm==v0.4.12
3-
torch==1.9.1+cu111
3+
torch==1.10.0+cu113

Diff for: ‎tests/py/requirements.txt

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
torchvision==0.10.1+cu111
2-
-f https://download.pytorch.org/whl/torch_stable.html
1+
torchvision==0.11.1+cu113
2+
-f https://download.pytorch.org/whl/torch_stable.html

0 commit comments

Comments
 (0)