Skip to content

Commit

Permalink
Update setup_env_gpu_cuda.bat
Browse files Browse the repository at this point in the history
  • Loading branch information
jchen351 committed Nov 8, 2023
1 parent ae609f8 commit 8af7f20
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down

0 comments on commit 8af7f20

Please sign in to comment.