From 43dc530b77ac9db4e7650e08064dacf906bc9252 Mon Sep 17 00:00:00 2001 From: chengang Date: Mon, 15 Mar 2021 17:49:18 +0800 Subject: [PATCH] remove hardcode cuda path --- mindspore/ccsrc/CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/CMakeLists.txt b/mindspore/ccsrc/CMakeLists.txt index 96c595a8a3..ed6a608a5c 100644 --- a/mindspore/ccsrc/CMakeLists.txt +++ b/mindspore/ccsrc/CMakeLists.txt @@ -355,7 +355,7 @@ if(ENABLE_D) set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling) elseif(ENABLE_GPU) - set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/cuda/lib64) + set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:${CUDA_PATH}/lib64) endif() set(MINDSPORE_RPATH ${ORIGIN_PATH}/lib:${MINDSPORE_RPATH}) set_target_properties(_c_expression PROPERTIES INSTALL_RPATH ${MINDSPORE_RPATH})