From 87975bdeef40ab79cb3f9d287e74af056cb8ef9c Mon Sep 17 00:00:00 2001 From: satyajandhyala Date: Mon, 2 Aug 2021 09:18:44 -0700 Subject: [PATCH] Use CUDA_HOME and CUDNN_HOME from the environment if they are not specified on the command line. (#8575) --- tools/ci_build/build.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/ci_build/build.py b/tools/ci_build/build.py index bd0a178a06..85ab066140 100644 --- a/tools/ci_build/build.py +++ b/tools/ci_build/build.py @@ -782,8 +782,8 @@ def generate_build_tree(cmake_path, source_dir, build_dir, cuda_home, cudnn_home add_cmake_define_without_override(cmake_extra_defines, "onnxruntime_USE_CUDA", "ON") add_cmake_define_without_override(cmake_extra_defines, "onnxruntime_CUDA_VERSION", args.cuda_version) # TODO: this variable is not really needed - add_cmake_define_without_override(cmake_extra_defines, "onnxruntime_CUDA_HOME", args.cuda_home) - add_cmake_define_without_override(cmake_extra_defines, "onnxruntime_CUDNN_HOME", args.cudnn_home) + add_cmake_define_without_override(cmake_extra_defines, "onnxruntime_CUDA_HOME", cuda_home) + add_cmake_define_without_override(cmake_extra_defines, "onnxruntime_CUDNN_HOME", cudnn_home) if is_windows(): if args.enable_msvc_static_runtime: