diff --git a/CMakeLists.txt b/CMakeLists.txt
index 358334c2..0ca9fcee 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -75,7 +75,11 @@ elseif(DEFINED ENV{D_LINK_PATH})
     find_library(resource libresource.so ${GE_LIB_PATH})
 else()
     # Ascend mode
-    set(HIAI_INSTALLED_DIR /usr/local/Ascend)
+    if(DEFINED ENV{ASCEND_CUSTOM_PATH})
+        set(HIAI_INSTALLED_DIR $ENV{D_LINK_PATH})
+    else()
+        set(HIAI_INSTALLED_DIR /usr/local/Ascend)
+    endif()
     set(HIAI_DRIVER_DIR ${HIAI_INSTALLED_DIR}/driver/lib64/common)
     set(HIAI_RUNTIME_DIR ${HIAI_INSTALLED_DIR}/fwkacllib/lib64)
     find_library(c_sec libc_sec.so ${HIAI_DRIVER_DIR})