Skip to content

Commit aa86623

Browse files
yf711guschmue
authored andcommitted
[EP Perf] Update env to ubuntu 22.04 (#23570)
### Description <!-- Describe your changes. --> * Update env to cuda 12.6/ubuntu 22.04 (ubuntu 20.04 uses outdated py38 by default) * Clean old trt8.6 test config ### 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 0524d10 commit aa86623

File tree

7 files changed

+6
-199
lines changed

7 files changed

+6
-199
lines changed

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

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,8 @@
1616
import sys
1717

1818
TRT_DOCKER_FILES = {
19-
"8.6_cuda11.8_cudnn8": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_8_tensorrt8_6",
20-
"8.6_cuda12.3_cudnn9": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_3_tensorrt8_6",
2119
"10.7_cuda11.8_cudnn8": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda11_tensorrt10",
22-
"10.7_cuda12.5_cudnn9": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_tensorrt10",
20+
"10.7_cuda12.6_cudnn9": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_tensorrt10",
2321
"BIN": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_tensorrt_bin",
2422
}
2523

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

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,10 @@ parameters:
88
- name: TrtVersion
99
displayName: TensorRT Version
1010
type: string
11-
default: 10.7_cuda12.5_cudnn9
11+
default: 10.7_cuda12.6_cudnn9
1212
values:
13-
- 8.6_cuda11.8_cudnn8
14-
- 8.6_cuda12.3_cudnn9
1513
- 10.7_cuda11.8_cudnn8
16-
- 10.7_cuda12.5_cudnn9
14+
- 10.7_cuda12.6_cudnn9
1715
- BIN
1816

1917
- name: UseTensorrtOssParser

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

Lines changed: 0 additions & 93 deletions
This file was deleted.

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
# Dockerfile to run ONNXRuntime with TensorRT integration
66

77
# Build base image with required system packages
8-
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04 AS base
8+
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 AS base
99

1010
# The local directory into which to build and install CMAKE
1111
ARG ONNXRUNTIME_LOCAL_CODE_DIR=/code

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

Lines changed: 0 additions & 96 deletions
This file was deleted.

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
# Dockerfile to run ONNXRuntime with TensorRT integration
66

77
# Build base image with required system packages
8-
FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu20.04 AS base
8+
FROM nvidia/cuda:12.6.3-cudnn-devel-ubuntu22.04 AS base
99

1010
# The local directory into which to build and install CMAKE
1111
ARG ONNXRUNTIME_LOCAL_CODE_DIR=/code

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
# Dockerfile to run ONNXRuntime with TensorRT installed from provided binaries
66

77
# Build base image with required system packages
8-
FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu20.04 AS base
8+
FROM nvidia/cuda:12.6.3-cudnn-devel-ubuntu22.04 AS base
99

1010
# The local directory into which to build and install CMAKE
1111
ARG ONNXRUNTIME_LOCAL_CODE_DIR=/code

0 commit comments

Comments
 (0)