Use TRT_VERSION=10.4.0.26-1.cuda12.6

microsoft · Sep 12, 2024 · 27d77b8 · 27d77b8
1 parent 7090cc4
commit 27d77b8
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/tools/ci_build/github/azure-pipelines/nuget/templates/test_linux.yml b/tools/ci_build/github/azure-pipelines/nuget/templates/test_linux.yml
@@ -61,7 +61,7 @@ stages:
           ${{ if eq(parameters.CudaVersion, '12.2') }}:
             DockerBuildArgs: "
             --build-arg BASEIMAGE=nvidia/cuda:12.2.2-devel-ubuntu20.04
-            --build-arg TRT_VERSION=10.4.0.26-1+cuda12.5
+            --build-arg TRT_VERSION=10.4.0.26-1+cuda12.6
             --build-arg BUILD_UID=$( id -u )
             "
           ${{ else }}:

diff --git a/tools/ci_build/github/azure-pipelines/stages/jobs/py-linux-cuda-package-test-job.yml b/tools/ci_build/github/azure-pipelines/stages/jobs/py-linux-cuda-package-test-job.yml
@@ -51,7 +51,7 @@ jobs:
         ${{ if eq(parameters.CudaVersion, '11.8') }}:
           value: 10.4.0.26-1.cuda11.8
         ${{ if eq(parameters.CudaVersion, '12.2') }}:
-          value: 10.4.0.26-1.cuda12.5
+          value: 10.4.0.26-1.cuda12.6
     pool: ${{ parameters.machine_pool }}
     steps:
       - checkout: self

diff --git a/tools/ci_build/github/azure-pipelines/stages/nuget-linux-cuda-packaging-stage.yml b/tools/ci_build/github/azure-pipelines/stages/nuget-linux-cuda-packaging-stage.yml
@@ -149,7 +149,7 @@ stages:
       ${{ if eq(parameters.CudaVersion, '11.8') }}:
         value: 10.4.0.26-1.cuda11.8
       ${{ if eq(parameters.CudaVersion, '12.2') }}:
-        value: 10.4.0.26-1.cuda12.5
+        value: 10.4.0.26-1.cuda12.6
     steps:
     - checkout: self                           # due to checkout multiple repos, the root directory is $(Build.SourcesDirectory)/onnxruntime
       submodules: false

diff --git a/tools/ci_build/github/azure-pipelines/stages/py-cuda-packaging-stage.yml b/tools/ci_build/github/azure-pipelines/stages/py-cuda-packaging-stage.yml
@@ -82,4 +82,4 @@ stages:
             trt_version: 10.4.0.26-1.cuda11.8
           ${{ if eq(parameters.cuda_version, '12.2') }}:
             docker_base_image: onnxruntimebuildcache.azurecr.io/internal/azureml/onnxruntime/build/cuda12_x64_ubi8_gcc12:20240719.1
-            trt_version: 10.4.0.26-1.cuda12.5
+            trt_version: 10.4.0.26-1.cuda12.6
diff --git a/tools/ci_build/github/azure-pipelines/templates/py-linux-gpu.yml b/tools/ci_build/github/azure-pipelines/templates/py-linux-gpu.yml
@@ -25,7 +25,7 @@ parameters:
   default: '10.4.0.26-1.cuda11.8'
   values:
     - 10.4.0.26-1.cuda11.8
-    - 10.4.0.26-1.cuda12.5
+    - 10.4.0.26-1.cuda12.6
 - name: cuda_version
   type: string
   default: '11.8'

diff --git a/tools/ci_build/github/azure-pipelines/templates/py-packaging-linux-test-cuda.yml b/tools/ci_build/github/azure-pipelines/templates/py-packaging-linux-test-cuda.yml
@@ -21,7 +21,7 @@ parameters:
   default: '10.4.0.26-1.cuda11.8'
   values:
     - 10.4.0.26-1.cuda11.8
-    - 10.4.0.26-1.cuda12.5
+    - 10.4.0.26-1.cuda12.6
 - name: cuda_version
   type: string
   default: '11.8'

diff --git a/tools/ci_build/github/linux/docker/Dockerfile.package_ubi8_cuda_tensorrt10_0 b/tools/ci_build/github/linux/docker/Dockerfile.package_ubi8_cuda_tensorrt10_0
@@ -6,7 +6,7 @@
 
 # Build base image with required system packages
 ARG BASEIMAGE=nvidia/cuda:12.5.1-cudnn-devel-ubi8
-ARG TRT_VERSION=10.4.0.26-1.cuda12.4
+ARG TRT_VERSION=10.4.0.26-1.cuda12.6
 FROM $BASEIMAGE AS base
 ARG TRT_VERSION
 ENV PATH /opt/python/cp38-cp38/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/src/tensorrt/bin:${PATH}