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

[Fix] Python Packaging Pipeline exception. (#15568)

Browse source 
### Description
supplement of #15299

### Motivation and Context
It broke Python Packaging Pipeline since April 12.
This commit is contained in:
Yi Zhang 2023-04-19 21:57:14 +08:00 committed by GitHub
parent 59ea35d592
commit 573e4cf95f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -34,6 +34,7 @@ jobs:
inputs:
targetType: filePath
filePath: tools/ci_build/github/linux/run_python_dockerbuild.sh
# please check ONNXRUNTIME_CUDA_VERSION in tools/ci_build/github/linux/build_linux_arm64_python_package.sh
arguments: -i onnxruntimecuda118xtrt86build${{ parameters.arch }} -x "-d GPU"
- task: PublishBuildArtifacts@1

2
tools/ci_build/github/linux/build_linux_arm64_python_package.sh
View file

@ -42,7 +42,7 @@ fi
if [ "$BUILD_DEVICE" == "GPU" ]; then
#Enable CUDA and TRT EPs.
ONNXRUNTIME_CUDA_VERSION="11.6"
ONNXRUNTIME_CUDA_VERSION="11.8"
BUILD_ARGS+=("--use_cuda" "--use_tensorrt" "--cuda_version=$ONNXRUNTIME_CUDA_VERSION" "--tensorrt_home=/usr" "--cuda_home=/usr/local/cuda-$ONNXRUNTIME_CUDA_VERSION" "--cudnn_home=/usr/local/cuda-$ONNXRUNTIME_CUDA_VERSION" "--cmake_extra_defines" "CMAKE_CUDA_ARCHITECTURES=52;60;61;70;75;80")
elif [ "$BUILD_DEVICE" == "AZURE" ]; then
BUILD_ARGS+=("--use_azure")