saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Change cuda image for tensorRT to the one with cudnn8 (#18102)

Browse source 
### Description
copilot:summary


### Motivation and Context
copliot::walkthrough
This commit is contained in:
Jian Chen 2023-10-26 16:28:57 -07:00 committed by GitHub
parent f2e19a8ccf
commit 7c18c60bc2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 7 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tools/ci_build/github/azure-pipelines/c-api-noopenmp-packaging-pipelines.yml
View file

@ -490,7 +490,7 @@ stages:
tools/ci_build/get_docker_image.py \
--dockerfile tools/ci_build/github/linux/docker/Dockerfile.manylinux2_28_cuda \
--context tools/ci_build/github/linux/docker \
--docker-build-args "--network=host --build-arg BASEIMAGE=nvidia/cuda:11.8.0-devel-ubi8 --build-arg INSTALL_CUDNN=true --build-arg TRT_VERSION=8.6.1.6-1.cuda11.8 --build-arg BUILD_UID=$( id -u )" \
--docker-build-args "--network=host --build-arg BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubi8 --build-arg TRT_VERSION=8.6.1.6-1.cuda11.8 --build-arg BUILD_UID=$( id -u )" \
--container-registry onnxruntimebuildcache \
--multiple_repos \
--repository onnxruntimecuda118xtrt86build

4
tools/ci_build/github/azure-pipelines/linux-gpu-ci-pipeline.yml
View file

@ -62,9 +62,8 @@ jobs:
Context: tools/ci_build/github/linux/docker
DockerBuildArgs: "
--network=host
--build-arg BASEIMAGE=nvidia/cuda:11.8.0-devel-ubi8
--build-arg BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubi8
--build-arg TRT_VERSION=8.6.1.6-1.cuda11.8
--build-arg INSTALL_CUDNN=true
--build-arg BUILD_UID=$( id -u )
"
Repository: onnxruntimecuda11build
@ -166,7 +165,6 @@ jobs:
--network=host
--build-arg BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubi8
--build-arg TRT_VERSION=8.6.1.6-1.cuda11.8
--build-arg INSTALL_CUDNN=true
--build-arg BUILD_UID=$( id -u )
"
Repository: onnxruntimecuda11build

1
tools/ci_build/github/azure-pipelines/linux-gpu-tensorrt-ci-pipeline.yml
View file

@ -63,7 +63,6 @@ jobs:
--network=host
--build-arg BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubi8
--build-arg TRT_VERSION=8.6.1.6-1.cuda11.8
--build-arg INSTALL_CUDNN=true
--build-arg BUILD_UID=$( id -u )
"
Repository: onnxruntimetensorrt86gpubuild

3
tools/ci_build/github/azure-pipelines/templates/linux-gpu-tensorrt-packaging-pipeline.yml
View file

@ -48,9 +48,8 @@ stages:
Context: tools/ci_build/github/linux/docker
DockerBuildArgs: "
--network=host
--build-arg BASEIMAGE=nvidia/cuda:11.8.0-devel-ubi8
--build-arg BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubi8
--build-arg TRT_VERSION=8.6.1.6-1.cuda11.8
--build-arg INSTALL_CUDNN=true
--build-arg BUILD_UID=$( id -u )
"
Repository: onnxruntimecuda118xtrt86build

3
tools/ci_build/github/azure-pipelines/templates/py-linux-gpu.yml
View file

@ -40,9 +40,8 @@ jobs:
Context: tools/ci_build/github/linux/docker
DockerBuildArgs: "
--network=host
--build-arg BASEIMAGE=nvidia/cuda:11.8.0-devel-ubi8
--build-arg BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubi8
--build-arg TRT_VERSION=8.6.1.6-1.cuda11.8
--build-arg INSTALL_CUDNN=true
--build-arg BUILD_UID=$( id -u )
--build-arg PLATFORM=${{ parameters.arch }}
"

3
tools/ci_build/github/azure-pipelines/templates/py-packaging-linux-test-cuda.yml
View file

@ -85,9 +85,8 @@ jobs:
Context: tools/ci_build/github/linux/docker
DockerBuildArgs: "
--network=host
--build-arg BASEIMAGE=nvidia/cuda:11.8.0-devel-ubi8
--build-arg BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubi8
--build-arg TRT_VERSION=8.6.1.6-1.cuda11.8
--build-arg INSTALL_CUDNN=true
--build-arg BUILD_UID=$( id -u )
--build-arg PLATFORM=${{ parameters.arch }}
"

14
tools/ci_build/github/linux/docker/Dockerfile.manylinux2_28_cuda
View file

@ -1,14 +1,13 @@
# The default ARGs are for cuda 12.2 with TRT v = 8.6.1.6-1.cuda12.0
# The default ARGs are for cuda 11.8 with cudnn8,TensorRT is optional
# Please overwirete BASEIMAGE, TRT_VERSION and other arguments with
# --docker-build-args ' --build-arg BASEIMAGE=other_base_image --build-arg TRT_VERSION=other_trt_version etc...'
# for other cuda version and TRT version
ARG POLICY=manylinux_2_28
ARG PLATFORM=x86_64
ARG BASEIMAGE=nvidia/cuda:12.2.0-devel-ubi8
ARG BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubi8
ARG DEVTOOLSET_ROOTPATH=/usr
ARG LD_LIBRARY_PATH_ARG=/usr/local/lib64
ARG PREPEND_PATH=/usr/local/cuda/binet
ARG INSTALL_CUDNN=false
#Build manylinux docker image begin
FROM $BASEIMAGE AS runtime_base
@ -154,15 +153,6 @@ CMD ["/bin/bash"]
#Build manylinux docker image end
#Install optinal Cudnn
RUN if [ "$INSTALL_CUDNN" = true ]; then \
CUDA_VERSION=$(nvcc --version | sed -n 's/^.*release \([0-9]\+\.[0-9]\+\).*$/\1/p') && \
dnf -y install \
libcudnn8-devel-*cuda${CUDA_VERSION}* \
libcudnn8-*cuda${CUDA_VERSION}* ; \
fi
#Install TensorRT only if TRT_VERSION is not empty
RUN if [ -n "$TRT_VERSION" ]; then \
echo "TRT_VERSION is $TRT_VERSION" && \