Skip to content

Commit c19791f

Browse files
lanluo-nvidialanluo-nvidia
authored
cherry-pick 3686: upgrade tensorrt from 10.11 to 10.12 (#3690)
1 parent 1ecd7ef commit c19791f

File tree

6 files changed

+27
-19
lines changed

6 files changed

+27
-19
lines changed

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,10 @@
5252
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11.0/zip/TensorRT-10.11.0.33.Windows.win10.cuda-12.9.zip",
5353
"strip_prefix": "TensorRT-10.11.0.33",
5454
},
55+
"10.12.0": {
56+
"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",
57+
"strip_prefix": "TensorRT-10.12.0.36",
58+
},
5559
},
5660
"linux": {
5761
"10.3.0": {
@@ -78,6 +82,10 @@
7882
"urls": "https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11.0/tars/TensorRT-10.11.0.33.Linux.x86_64-gnu.cuda-12.9.tar.gz",
7983
"strip_prefix": "TensorRT-10.11.0.33",
8084
},
85+
"10.12.0": {
86+
"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",
87+
"strip_prefix": "TensorRT-10.12.0.36",
88+
},
8189
},
8290
}
8391

‎MODULE.bazel‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -103,18 +103,18 @@ http_archive(
103103
http_archive(
104104
name = "tensorrt",
105105
build_file = "@//third_party/tensorrt/archive:BUILD",
106-
strip_prefix = "TensorRT-10.11.0.33",
106+
strip_prefix = "TensorRT-10.12.0.36",
107107
urls = [
108-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11.0/tars/TensorRT-10.11.0.33.Linux.x86_64-gnu.cuda-12.9.tar.gz",
108+
"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",
109109
],
110110
)
111111

112112
http_archive(
113113
name = "tensorrt_sbsa",
114114
build_file = "@//third_party/tensorrt/archive:BUILD",
115-
strip_prefix = "TensorRT-10.11.0.33",
115+
strip_prefix = "TensorRT-10.12.0.36",
116116
urls = [
117-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11.0/tars/TensorRT-10.11.0.33.Linux.aarch64-gnu.cuda-12.9.tar.gz",
117+
"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",
118118
],
119119
)
120120

@@ -130,9 +130,9 @@ http_archive(
130130
http_archive(
131131
name = "tensorrt_win",
132132
build_file = "@//third_party/tensorrt/archive:BUILD",
133-
strip_prefix = "TensorRT-10.11.0.33",
133+
strip_prefix = "TensorRT-10.12.0.36",
134134
urls = [
135-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11.0/zip/TensorRT-10.11.0.33.Windows.win10.cuda-12.9.zip",
135+
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.12.0/zip/TensorRT-10.12.0.36.Windows.win10.cuda-12.9.zip",
136136
],
137137
)
138138

‎dev_dep_versions.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
11
__cuda_version__: "12.8"
2-
__tensorrt_version__: "10.11.0"
2+
__tensorrt_version__: "10.12.0"

‎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.11.0.33
18+
- The example below uses TensorRT 10.12.0.36
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.11.0 -f docker/Dockerfile -t torch_tensorrt:latest .
25+
DOCKER_BUILDKIT=1 docker build --build-arg TENSORRT_VERSION=10.12.0 -f docker/Dockerfile -t torch_tensorrt:latest .
2626
```
2727

2828
Run:

‎pyproject.toml‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,10 +62,10 @@ dependencies = [
6262
"torch>=2.8.0,<2.9.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
6363
# "torch>=2.7.0,<2.8.0; platform_machine == 'aarch64' and 'tegra' in platform_release",
6464

65-
"tensorrt>=10.11.0,<10.12.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
66-
"tensorrt-cu12>=10.11.0,<10.12.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
67-
"tensorrt-cu12-bindings>=10.11.0,<10.12.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
68-
"tensorrt-cu12-libs>=10.11.0,<10.12.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
65+
"tensorrt>=10.12.0,<10.13.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
66+
"tensorrt-cu12>=10.12.0,<10.13.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
67+
"tensorrt-cu12-bindings>=10.12.0,<10.13.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
68+
"tensorrt-cu12-libs>=10.12.0,<10.13.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
6969

7070
"tensorrt>=10.3.0,<10.4.0;platform_machine == 'aarch64' and 'tegra' in platform_release",
7171
"tensorrt-cu12>=10.3.0,<10.4.0; platform_machine == 'aarch64' and 'tegra' in platform_release",

‎toolchains/ci_workspaces/MODULE.bazel.tmpl‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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.11.0.33",
78+
strip_prefix = "TensorRT-10.12.0.36",
7979
urls = [
80-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11.0/tars/TensorRT-10.11.0.33.Linux.x86_64-gnu.cuda-12.9.tar.gz",
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",
8181
],
8282
)
8383

8484
http_archive(
8585
name = "tensorrt_sbsa",
8686
build_file = "@//third_party/tensorrt/archive:BUILD",
87-
strip_prefix = "TensorRT-10.11.0.33",
87+
strip_prefix = "TensorRT-10.12.0.36",
8888
urls = [
89-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11.0/tars/TensorRT-10.11.0.33.Linux.aarch64-gnu.cuda-12.9.tar.gz",
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",
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.11.0.33",
105+
strip_prefix = "TensorRT-10.12.0.36",
106106
urls = [
107-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.11.0/zip/TensorRT-10.11.0.33.Windows.win10.cuda-12.9.zip",
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",
108108
],
109109
)
110110

0 commit comments

Comments
 (0)