Skip to content

Commit a93d5e9

Browse files
peri044peri044
committed
chore: Upgrade TRT to 10.13.2.6
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 40d4d41 commit a93d5e9

File tree

6 files changed

+31
-26
lines changed

6 files changed

+31
-26
lines changed

‎.github/scripts/generate-tensorrt-test-matrix.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,9 @@
1111
# channel: nightly if the future tensorRT version test workflow is triggered from the main branch or your personal branch
1212
# channel: test if the future tensorRT version test workflow is triggered from the release branch(release/2.5 etc....)
1313
CUDA_VERSIONS_DICT = {
14-
"nightly": ["cu128"],
15-
"test": ["cu118", "cu126", "cu128"],
16-
"release": ["cu118", "cu126", "cu128"],
14+
"nightly": ["cu129"],
15+
"test": ["cu126", "cu128", "cu129"],
16+
"release": ["cu126", "cu128", "cu129"],
1717
}
1818

1919
# please update the python version you want to test with the future tensorRT version here
@@ -56,6 +56,10 @@
5656
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.12.0/zip/TensorRT-10.12.0.36.Windows.win10.cuda-12.9.zip",
5757
"strip_prefix": "TensorRT-10.12.0.36",
5858
},
59+
"10.13.2": {
60+
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.2/zip/TensorRT-10.13.2.6.Windows.win10.cuda-12.9.zip",
61+
"strip_prefix": "TensorRT-10.13.2.6",
62+
},
5963
},
6064
"linux": {
6165
"10.3.0": {
@@ -86,6 +90,10 @@
8690
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.12.0/tars/TensorRT-10.12.0.36.Linux.x86_64-gnu.cuda-12.9.tar.gz",
8791
"strip_prefix": "TensorRT-10.12.0.36",
8892
},
93+
"10.13.2": {
94+
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.2/tars/TensorRT-10.13.2.6.Linux.x86_64-gnu.cuda-12.9.tar.gz",
95+
"strip_prefix": "TensorRT-10.13.2.6",
96+
},
8997
},
9098
}
9199

‎MODULE.bazel

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,6 @@ git_override(
2222
remote = "https://github.com/narendasan/rules_pkg",
2323
)
2424

25-
local_repository = use_repo_rule("@bazel_tools//tools/build_defs/repo:local.bzl", "local_repository")
26-
27-
2825
new_local_repository = use_repo_rule("@bazel_tools//tools/build_defs/repo:local.bzl", "new_local_repository")
2926

3027
# External dependency for torch_tensorrt if you already have precompiled binaries.
@@ -104,18 +101,18 @@ http_archive(
104101
http_archive(
105102
name = "tensorrt",
106103
build_file = "@//third_party/tensorrt/archive:BUILD",
107-
strip_prefix = "TensorRT-10.12.0.36",
104+
strip_prefix = "TensorRT-10.13.2.6",
108105
urls = [
109-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.12.0/tars/TensorRT-10.12.0.36.Linux.x86_64-gnu.cuda-12.9.tar.gz",
106+
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.2/tars/TensorRT-10.13.2.6.Linux.x86_64-gnu.cuda-12.9.tar.gz",
110107
],
111108
)
112109

113110
http_archive(
114111
name = "tensorrt_sbsa",
115112
build_file = "@//third_party/tensorrt/archive:BUILD",
116-
strip_prefix = "TensorRT-10.12.0.36",
113+
strip_prefix = "TensorRT-10.13.2.6",
117114
urls = [
118-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.12.0/tars/TensorRT-10.12.0.36.Linux.aarch64-gnu.cuda-12.9.tar.gz",
115+
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.2/tars/TensorRT-10.13.2.6.Linux.aarch64-gnu.cuda-13.0.tar.gz",
119116
],
120117
)
121118

@@ -131,9 +128,9 @@ http_archive(
131128
http_archive(
132129
name = "tensorrt_win",
133130
build_file = "@//third_party/tensorrt/archive:BUILD",
134-
strip_prefix = "TensorRT-10.12.0.36",
131+
strip_prefix = "TensorRT-10.13.2.6",
135132
urls = [
136-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.12.0/zip/TensorRT-10.12.0.36.Windows.win10.cuda-12.9.zip",
133+
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.2/zip/TensorRT-10.13.2.6.Windows.win10.cuda-12.9.zip",
137134
],
138135
)
139136

‎README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ Torch-TensorRT
77
[![Documentation](https://img.shields.io/badge/docs-master-brightgreen)](https://nvidia.github.io/Torch-TensorRT/)
88
[![pytorch](https://img.shields.io/badge/PyTorch-2.8-green)](https://download.pytorch.org/whl/nightly/cu128)
99
[![cuda](https://img.shields.io/badge/CUDA-12.8-green)](https://developer.nvidia.com/cuda-downloads)
10-
[![trt](https://img.shields.io/badge/TensorRT-10.12.0-green)](https://github.com/nvidia/tensorrt-llm)
10+
[![trt](https://img.shields.io/badge/TensorRT-10.13.2.6-green)](https://github.com/nvidia/tensorrt-llm)
1111
[![license](https://img.shields.io/badge/license-BSD--3--Clause-blue)](./LICENSE)
1212
[![Linux x86-64 Nightly Wheels](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-x86_64.yml/badge.svg?branch=nightly)](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-x86_64.yml)
1313
[![Linux SBSA Nightly Wheels](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-aarch64.yml/badge.svg?branch=nightly)](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-aarch64.yml)
@@ -123,7 +123,7 @@ These are the following dependencies used to verify the testcases. Torch-TensorR
123123
- Bazel 8.1.1
124124
- Libtorch 2.9.0.dev (latest nightly)
125125
- CUDA 12.8 (CUDA 12.6 on Jetson)
126-
- TensorRT 10.12 (TensorRT 10.3 on Jetson)
126+
- TensorRT 10.13.2.6 (TensorRT 10.3 on Jetson)
127127
128128
## Deprecation Policy
129129

‎docker/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,14 +15,14 @@
1515

1616
### Instructions
1717

18-
- The example below uses TensorRT 10.12.0.36
18+
- The example below uses TensorRT 10.13.2.6
1919
- See <a href="https://github.com/pytorch/TensorRT#dependencies">dependencies</a> for a list of current default dependencies.
2020

2121
> From root of Torch-TensorRT repo
2222
2323
Build:
2424
```
25-
DOCKER_BUILDKIT=1 docker build --build-arg TENSORRT_VERSION=10.12.0 -f docker/Dockerfile -t torch_tensorrt:latest .
25+
DOCKER_BUILDKIT=1 docker build --build-arg TENSORRT_VERSION=10.13.2 -f docker/Dockerfile -t torch_tensorrt:latest .
2626
```
2727

2828
Run:

‎pyproject.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,9 @@ dependencies = [
5454
"torch>=2.9.0.dev,<2.10.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
5555
"torch>=2.8.0,<2.9.0; platform_machine == 'aarch64' and 'tegra' in platform_release",
5656

57-
"tensorrt>=10.12.0,<10.13.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
58-
"tensorrt-cu12-bindings>=10.12.0,<10.13.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
59-
"tensorrt-cu12-libs>=10.12.0,<10.13.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
57+
"tensorrt>=10.13.0,<10.14.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
58+
"tensorrt-cu12-bindings>=10.13.0,<10.14.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
59+
"tensorrt-cu12-libs>=10.13.0,<10.14.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
6060

6161
"tensorrt>=10.3.0,<10.4.0;platform_machine == 'aarch64' and 'tegra' in platform_release",
6262
"packaging>=23",

‎toolchains/ci_workspaces/MODULE.bazel.tmpl

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ new_local_repository(
4545
new_local_repository(
4646
name = "cuda_l4t",
4747
build_file = "@//third_party/cuda:BUILD",
48-
path = "/usr/local/cuda-12.6",
48+
path = "/usr/local/cuda-12.9",
4949
)
5050

5151
new_local_repository(
@@ -75,18 +75,18 @@ http_archive = use_repo_rule("@bazel_tools//tools/build_defs/repo:http.bzl", "ht
7575
http_archive(
7676
name = "tensorrt",
7777
build_file = "@//third_party/tensorrt/archive:BUILD",
78-
strip_prefix = "TensorRT-10.12.0.36",
78+
strip_prefix = "TensorRT-10.13.2.6",
7979
urls = [
80-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.12.0/tars/TensorRT-10.12.0.36.Linux.x86_64-gnu.cuda-12.9.tar.gz",
80+
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.2/tars/TensorRT-10.13.2.6.Linux.x86_64-gnu.cuda-12.9.tar.gz",
8181
],
8282
)
8383

8484
http_archive(
8585
name = "tensorrt_sbsa",
8686
build_file = "@//third_party/tensorrt/archive:BUILD",
87-
strip_prefix = "TensorRT-10.12.0.36",
87+
strip_prefix = "TensorRT-10.13.2.6",
8888
urls = [
89-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.12.0/tars/TensorRT-10.12.0.36.Linux.aarch64-gnu.cuda-12.9.tar.gz",
89+
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.2/tars/TensorRT-10.13.2.6.Linux.aarch64-gnu.cuda-13.0.tar.gz",
9090
],
9191
)
9292

@@ -102,9 +102,9 @@ http_archive(
102102
http_archive(
103103
name = "tensorrt_win",
104104
build_file = "@//third_party/tensorrt/archive:BUILD",
105-
strip_prefix = "TensorRT-10.12.0.36",
105+
strip_prefix = "TensorRT-10.13.2.6",
106106
urls = [
107-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.12.0/zip/TensorRT-10.12.0.36.Windows.win10.cuda-12.9.zip",
107+
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.2/zip/TensorRT-10.13.2.6.Windows.win10.cuda-12.9.zip",
108108
],
109109
)
110110

0 commit comments

Comments
 (0)