update ep_perf

microsoft · Dec 5, 2024 · 5866881 · 5866881
1 parent d381f1c
commit 5866881
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/onnxruntime/python/tools/tensorrt/perf/build/build_image.py b/onnxruntime/python/tools/tensorrt/perf/build/build_image.py
@@ -15,10 +15,10 @@
 from typing import List, Optional
 
 TRT_DOCKER_FILES = {
-    "8.6.cuda_11_8_cudnn_8": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_8_tensorrt8_6",
-    "8.6.cuda_12_3_cudnn_9": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_3_tensorrt8_6",
-    "10.5.cuda_11_8_cudnn_8": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_tensorrt10",
-    "10.5.cuda_12_5_cudnn_9": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_tensorrt10",
+    "8.6(cuda11.8+cudnn8)": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_8_tensorrt8_6",
+    "8.6(cuda12.3+cudnn9)": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_3_tensorrt8_6",
+    "10.7(cuda11.8+cudnn8)": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_tensorrt10",
+    "10.7(cuda12+cudnn9)": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_tensorrt10",
     "BIN": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_tensorrt_bin",
 }
 

diff --git a/tools/ci_build/github/azure-pipelines/linux-gpu-tensorrt-daily-perf-pipeline.yml b/tools/ci_build/github/azure-pipelines/linux-gpu-tensorrt-daily-perf-pipeline.yml
@@ -8,12 +8,12 @@ parameters:
 - name: TrtVersion
   displayName: TensorRT Version
   type: string
-  default: 10.5.cuda_12_5_cudnn_9
+  default: 10.7(cuda12+cudnn9)
   values:
-  - 8.6.cuda_11_8_cudnn_8
-  - 8.6.cuda_12_3_cudnn_9
-  - 10.5.cuda_11_8_cudnn_8
-  - 10.5.cuda_12_5_cudnn_9
+  - 8.6(cuda11.8+cudnn8)
+  - 8.6(cuda12.3+cudnn9)
+  - 10.7(cuda11.8+cudnn8)
+  - 10.7(cuda12+cudnn9)
   - BIN
 
 - name: UseTensorrtOssParser

diff --git a/tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_tensorrt10 b/tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_tensorrt10
@@ -5,7 +5,7 @@
 # Dockerfile to run ONNXRuntime with TensorRT integration
 
 # Build base image with required system packages
-FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu20.04 AS base
+FROM nvidia/cuda:12.6.3-cudnn-devel-ubuntu20.04 AS base
 
 # The local directory into which to build and install CMAKE
 ARG ONNXRUNTIME_LOCAL_CODE_DIR=/code