mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-29 20:14:01 +00:00
test cu128 on ep_perf
This commit is contained in:
parent
15cea1bbf1
commit
92f06d8126
3 changed files with 4 additions and 4 deletions
|
|
@ -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",
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in a new issue