From 8af7f204b3c0c049f59040b1403fd29e9a20f86c Mon Sep 17 00:00:00 2001 From: Jian Chen Date: Wed, 8 Nov 2023 10:37:17 -0800 Subject: [PATCH] Update setup_env_gpu_cuda.bat --- .../azure-pipelines/templates/py-packaging-stage-cuda12.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage-cuda12.yml b/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage-cuda12.yml index 23db749627df6..b7f0b8f91a10d 100644 --- a/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage-cuda12.yml +++ b/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage-cuda12.yml @@ -89,7 +89,7 @@ stages: MACHINE_POOL: 'onnxruntime-Win2022-GPU-T4' PYTHON_VERSION: '3.11' EP_BUILD_FLAGS: --use_tensorrt --tensorrt_home=${{ variables.win_trt_home }} --cuda_home=${{ variables.win_cuda_home }} --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=52;60;61;70;75;80" - ENV_SETUP_SCRIPT: setup_env_gpu_cuda.bat + ENV_SETUP_SCRIPT: setup_env_gpu.bat EP_NAME: gpu