test cu128 on ep_perf

This commit is contained in:
yf711 2025-02-04 15:01:12 -08:00
parent 15cea1bbf1
commit 92f06d8126
3 changed files with 4 additions and 4 deletions

View file

@ -19,7 +19,7 @@ TRT_DOCKER_FILES = {
"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.5_cudnn9": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_tensorrt10",
"10.7_cuda12.8_cudnn9": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_cuda12_tensorrt10",
"BIN": "tools/ci_build/github/linux/docker/Dockerfile.ubuntu_tensorrt_bin",
}

View file

@ -8,12 +8,12 @@ parameters:
- name: TrtVersion
displayName: TensorRT Version
type: string
default: 10.7_cuda12.5_cudnn9
default: 10.7_cuda12.8_cudnn9
values:
- 8.6_cuda11.8_cudnn8
- 8.6_cuda12.3_cudnn9
- 10.7_cuda11.8_cudnn8
- 10.7_cuda12.5_cudnn9
- 10.7_cuda12.8_cudnn9
- BIN
- name: UseTensorrtOssParser

View file

@ -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.8.0-cudnn-devel-ubuntu20.04 AS base
# The local directory into which to build and install CMAKE
ARG ONNXRUNTIME_LOCAL_CODE_DIR=/code