Skip to content

Commit d9d07ad

Browse files
authored
[TensorRT EP] support TensorRT 10.7-GA (microsoft#23011)
### Description <!-- Describe your changes. --> Update CIs to TRT10.7 ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
1 parent a3bb3f1 commit d9d07ad

19 files changed

+33
-33
lines changed

cgmanifests/generated/cgmanifest.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@
196196
"component": {
197197
"type": "git",
198198
"git": {
199-
"commitHash": "bc0d2e35909b8456abe32f3b30a49bb0c125e8b7",
199+
"commitHash": "9c69a24bc2e20c8a511a4e6b06fd49639ec5300a",
200200
"repositoryUrl": "https://github.com/onnx/onnx-tensorrt.git"
201201
},
202202
"comments": "onnx_tensorrt"

cmake/deps.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,8 @@ microsoft_wil;https://github.com/microsoft/wil/archive/refs/tags/v1.0.230629.1.z
3636
mimalloc;https://github.com/microsoft/mimalloc/archive/refs/tags/v2.1.1.zip;d5ee7d34223d0567892db5179849939c8769dc41
3737
mp11;https://github.com/boostorg/mp11/archive/refs/tags/boost-1.82.0.zip;9bc9e01dffb64d9e0773b2e44d2f22c51aace063
3838
onnx;https://github.com/onnx/onnx/archive/refs/tags/v1.16.1.zip;2eb9198bb352757d5ff13977cbe0634898e0837c
39-
# Use the latest commit of 10.6-GA-ORT-DDS
40-
onnx_tensorrt;https://github.com/onnx/onnx-tensorrt/archive/bc0d2e35909b8456abe32f3b30a49bb0c125e8b7.zip;f233ae871ad82c023da62e5dd620639f00bc2d15
39+
# Use the latest commit of 10.7-GA
40+
onnx_tensorrt;https://github.com/onnx/onnx-tensorrt/archive/9c69a24bc2e20c8a511a4e6b06fd49639ec5300a.zip;ff1fe9af78eb129b4a4cdcb7450b7390b4436dd3
4141
protobuf;https://github.com/protocolbuffers/protobuf/archive/refs/tags/v21.12.zip;7cf2733949036c7d52fda017badcab093fe73bfa
4242
protoc_win64;https://github.com/protocolbuffers/protobuf/releases/download/v21.12/protoc-21.12-win64.zip;b4521f7ada5b260380f94c4bd7f1b7684c76969a
4343
protoc_win32;https://github.com/protocolbuffers/protobuf/releases/download/v21.12/protoc-21.12-win32.zip;3688010318192c46ce73213cdfb6b3e5656da874

onnxruntime/python/tools/tensorrt/perf/build/build_image.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,10 @@
1515
from typing import List, Optional
1616

1717
TRT_DOCKER_FILES = {
18-
"8.6.cuda_11_8_cudnn_8": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_8_tensorrt8_6",
19-
"8.6.cuda_12_3_cudnn_9": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_3_tensorrt8_6",
20-
"10.5.cuda_11_8_cudnn_8": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_tensorrt10",
21-
"10.5.cuda_12_5_cudnn_9": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_tensorrt10",
18+
"8.6_cuda11.8_cudnn8": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_8_tensorrt8_6",
19+
"8.6_cuda12.3_cudnn9": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_3_tensorrt8_6",
20+
"10.7_cuda11.8_cudnn8": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_tensorrt10",
21+
"10.7_cuda12.5_cudnn9": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_tensorrt10",
2222
"BIN": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_tensorrt_bin",
2323
}
2424

tools/ci_build/github/azure-pipelines/linux-gpu-tensorrt-daily-perf-pipeline.yml

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,12 @@ parameters:
88
- name: TrtVersion
99
displayName: TensorRT Version
1010
type: string
11-
default: 10.5.cuda_12_5_cudnn_9
11+
default: 10.7_cuda12.5_cudnn9
1212
values:
13-
- 8.6.cuda_11_8_cudnn_8
14-
- 8.6.cuda_12_3_cudnn_9
15-
- 10.5.cuda_11_8_cudnn_8
16-
- 10.5.cuda_12_5_cudnn_9
13+
- 8.6_cuda11.8_cudnn8
14+
- 8.6_cuda12.3_cudnn9
15+
- 10.7_cuda11.8_cudnn8
16+
- 10.7_cuda12.5_cudnn9
1717
- BIN
1818

1919
- name: UseTensorrtOssParser
@@ -198,4 +198,4 @@ jobs:
198198
parameters :
199199
condition : 'succeeded'
200200

201-
- template: templates/clean-agent-build-directory-step.yml
201+
- template: templates/clean-agent-build-directory-step.yml

tools/ci_build/github/azure-pipelines/py-cuda-alt-package-test-pipeline.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,6 @@ stages:
1919
python_wheel_suffix: '_gpu'
2020
timeout: 480
2121
docker_base_image: onnxruntimebuildcache.azurecr.io/internal/azureml/onnxruntime/build/cuda11_x64_almalinux8_gcc11:20241120.3
22-
trt_version: '10.6.0.26-1.cuda11.8'
22+
trt_version: '10.7.0.23-1.cuda11.8'
2323
cuda_version: '11.8'
2424

tools/ci_build/github/azure-pipelines/templates/common-variables.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
variables:
2-
common_trt_version: '10.6.0.26'
2+
common_trt_version: '10.7.0.23'
33
# As for Debian installation, replace '-1.' by '-1+' when assigning trt version below
44
linux_trt_version_cuda11: ${{ variables.common_trt_version }}-1.cuda11.8
55
linux_trt_version_cuda12: ${{ variables.common_trt_version }}-1.cuda12.6

tools/ci_build/github/azure-pipelines/templates/download-deps.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ steps:
1111
packageType: upack
1212
feed: '/7424c8e4-5c62-490e-95c4-79446f31017c'
1313
definition: '517c4f6f-5437-4392-a70d-4f15ec5be2f0'
14-
version: 1.0.201
14+
version: 1.0.202
1515
downloadPath: $(Build.BinariesDirectory)/deps
1616

1717
# The private ADO project
@@ -22,7 +22,7 @@ steps:
2222
packageType: upack
2323
feed: '/4c7631f5-24c0-4307-8822-1aa8f180c325'
2424
definition: 'fd9dd5ad-b73e-4678-890e-edcf680dbc1a'
25-
version: 1.0.201
25+
version: 1.0.202
2626
downloadPath: $(Build.BinariesDirectory)/deps
2727

2828
# You can add more ADO accounts at here.

tools/ci_build/github/azure-pipelines/templates/jobs/download_win_gpu_library.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@ parameters:
1313
- 12.2
1414
- name: TrtVersion
1515
type: string
16-
default: '10.6.0.26'
16+
default: '10.7.0.23'
1717
values:
1818
- 8.6.1.6
19-
- 10.6.0.26
19+
- 10.7.0.23
2020

2121
steps:
2222
- ${{ if eq(parameters.DownloadCUDA, true) }}:
@@ -42,7 +42,7 @@ steps:
4242
- powershell: |
4343
Write-Host "##vso[task.setvariable variable=trtCudaVersion;]12.0"
4444
displayName: Set trtCudaVersion
45-
- ${{ if and(eq(parameters.CudaVersion, '12.2'), eq(parameters.TrtVersion, '10.6.0.26')) }}:
45+
- ${{ if and(eq(parameters.CudaVersion, '12.2'), eq(parameters.TrtVersion, '10.7.0.23')) }}:
4646
- powershell: |
4747
Write-Host "##vso[task.setvariable variable=trtCudaVersion;]12.6"
4848
displayName: Set trtCudaVersion

tools/ci_build/github/azure-pipelines/templates/jobs/set-winenv.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,10 @@ parameters:
1515
default: '11.8'
1616
- name: win_trt_folder_cuda11
1717
type: string
18-
default: 'TensorRT-10.6.0.26.Windows10.x86_64.cuda-11.8'
18+
default: 'TensorRT-10.7.0.23.Windows10.x86_64.cuda-11.8'
1919
- name: win_trt_folder_cuda12
2020
type: string
21-
default: 'TensorRT-10.6.0.26.Windows10.x86_64.cuda-12.6'
21+
default: 'TensorRT-10.7.0.23.Windows10.x86_64.cuda-12.6'
2222

2323
steps:
2424
- ${{ if eq(parameters.DownloadCUDA, 'true') }}:

tools/ci_build/github/linux/docker/Dockerfile.package_ubi8_cuda_tensorrt10_0

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Build base image with required system packages
88
ARG BASEIMAGE=nvidia/cuda:12.5.1-cudnn-devel-ubi8
9-
ARG TRT_VERSION=10.6.0.26-1.cuda12.6
9+
ARG TRT_VERSION=10.7.0.23-1.cuda12.6
1010
FROM $BASEIMAGE AS base
1111
ARG TRT_VERSION
1212
ENV PATH=/opt/python/cp310-cp310/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/src/tensorrt/bin:${PATH}

tools/ci_build/github/linux/docker/Dockerfile.package_ubi8_cuda_tensorrt10_0_torch

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Build base image with required system packages
88
ARG BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubi8
9-
ARG TRT_VERSION=10.6.0.26-1.cuda11.8
9+
ARG TRT_VERSION=10.7.0.23-1.cuda11.8
1010
FROM $BASEIMAGE AS base
1111
ARG TRT_VERSION
1212
ENV PATH=/opt/python/cp310-cp310/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/src/tensorrt/bin:${PATH}

tools/ci_build/github/linux/docker/Dockerfile.package_ubuntu_2004_gpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Build base image with required system packages
88
ARG BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
9-
ARG TRT_VERSION=10.6.0.26-1+cuda11.8
9+
ARG TRT_VERSION=10.7.0.23-1+cuda11.8
1010
ARG LD_LIBRARY_PATH_ARG=/usr/local/lib64:/usr/local/cuda/lib64
1111
FROM $BASEIMAGE AS base
1212
ARG TRT_VERSION

tools/ci_build/github/linux/docker/Dockerfile.package_ubuntu_2204_gpu_ffmpeg

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Build base image with required system packages
88
ARG BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
9-
ARG TRT_VERSION=10.6.0.26-1+cuda11.8
9+
ARG TRT_VERSION=10.7.0.23-1+cuda11.8
1010
ARG LD_LIBRARY_PATH_ARG=/usr/local/lib64:/usr/local/cuda/lib64
1111
FROM $BASEIMAGE AS base
1212
ARG TRT_VERSION

tools/ci_build/github/linux/docker/Dockerfile.package_ubuntu_2204_gpu_opencv

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
# Build base image with required system packages
88
ARG BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
9-
ARG TRT_VERSION=10.6.0.26-1+cuda11.8
9+
ARG TRT_VERSION=10.7.0.23-1+cuda11.8
1010
ARG LD_LIBRARY_PATH_ARG=/usr/local/lib64:/usr/local/cuda/lib64
1111
FROM $BASEIMAGE AS base
1212
ARG TRT_VERSION

tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_tensorrt10

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ RUN pip install --upgrade pip
3131
RUN pip install psutil setuptools>=68.2.2
3232

3333
# Install TensorRT
34-
RUN TRT_VERSION="10.6.0.26-1+cuda11.8" &&\
34+
RUN TRT_VERSION="10.7.0.23-1+cuda11.8" &&\
3535
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub &&\
3636
apt-get update &&\
3737
apt-get install -y \

tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_tensorrt10

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ RUN pip install --upgrade pip
3131
RUN pip install setuptools>=68.2.2 psutil
3232

3333
# Install TensorRT
34-
RUN TRT_VERSION="10.6.0.26-1+cuda12.6" &&\
34+
RUN TRT_VERSION="10.7.0.23-1+cuda12.6" &&\
3535
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub &&\
3636
apt-get update &&\
3737
apt-get install -y \

tools/ci_build/github/linux/docker/inference/x86_64/python/cuda/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
ARG BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubi8
66

77
FROM $BASEIMAGE
8-
ARG TRT_VERSION=10.6.0.26-1.cuda11.8
8+
ARG TRT_VERSION=10.7.0.23-1.cuda11.8
99

1010
#Install TensorRT only if TRT_VERSION is not empty
1111
RUN if [ -n "${TRT_VERSION}" ]; then \

tools/ci_build/github/windows/setup_env_gpu.bat

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,10 @@ if exist PATH=%AGENT_TEMPDIRECTORY%\v12.2\ (
66
) else (
77
set PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\extras\CUPTI\lib64;%PATH%
88
)
9-
set PATH=%AGENT_TEMPDIRECTORY%\TensorRT-10.6.0.26.Windows10.x86_64.cuda-12.6\lib;%PATH%
9+
set PATH=%AGENT_TEMPDIRECTORY%\TensorRT-10.7.0.23.Windows10.x86_64.cuda-12.6\lib;%PATH%
1010

1111
@REM The default version is still cuda v12.2, because set cuda v11.8 after it
12-
set PATH=%PATH%;%AGENT_TEMPDIRECTORY%\TensorRT-10.6.0.26.Windows10.x86_64.cuda-11.8\lib
12+
set PATH=%PATH%;%AGENT_TEMPDIRECTORY%\TensorRT-10.7.0.23.Windows10.x86_64.cuda-11.8\lib
1313
if exist PATH=%AGENT_TEMPDIRECTORY%\v11.8\ (
1414
set PATH=%PATH%;%AGENT_TEMPDIRECTORY%\v11.8\bin;%AGENT_TEMPDIRECTORY%\v11.8\extras\CUPTI\lib64
1515
) else (

tools/ci_build/github/windows/setup_env_trt.bat

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,6 @@ if exist PATH=%AGENT_TEMPDIRECTORY%\v12.2\ (
66
) else (
77
set PATH=%PATH%;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\extras\CUPTI\lib64
88
)
9-
set PATH=%AGENT_TEMPDIRECTORY%\TensorRT-10.6.0.26.Windows10.x86_64.cuda-12.6\lib;%PATH%
9+
set PATH=%AGENT_TEMPDIRECTORY%\TensorRT-10.7.0.23.Windows10.x86_64.cuda-12.6\lib;%PATH%
1010
set GRADLE_OPTS=-Dorg.gradle.daemon=false
1111
set CUDA_MODULE_LOADING=LAZY

0 commit comments

Comments
 (0)