From 573e4cf95fba64452deccb21bdd8b923ca57146b Mon Sep 17 00:00:00 2001 From: Yi Zhang Date: Wed, 19 Apr 2023 21:57:14 +0800 Subject: [PATCH] [Fix] Python Packaging Pipeline exception. (#15568) ### Description supplement of #15299 ### Motivation and Context It broke Python Packaging Pipeline since April 12. --- .../ci_build/github/azure-pipelines/templates/py-linux-gpu.yml | 1 + tools/ci_build/github/linux/build_linux_arm64_python_package.sh | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/ci_build/github/azure-pipelines/templates/py-linux-gpu.yml b/tools/ci_build/github/azure-pipelines/templates/py-linux-gpu.yml index d2a5d8f6dd..d8a3619759 100644 --- a/tools/ci_build/github/azure-pipelines/templates/py-linux-gpu.yml +++ b/tools/ci_build/github/azure-pipelines/templates/py-linux-gpu.yml @@ -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 diff --git a/tools/ci_build/github/linux/build_linux_arm64_python_package.sh b/tools/ci_build/github/linux/build_linux_arm64_python_package.sh index 9af2f40a42..88ac111fd8 100755 --- a/tools/ci_build/github/linux/build_linux_arm64_python_package.sh +++ b/tools/ci_build/github/linux/build_linux_arm64_python_package.sh @@ -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")