fix: skip symlink when copy bin to |${NV_DIR}|. (#15)

* fix: skip symlink when copy bin to |${NV_DIR}|.

* fix: use |${FIND_BASE}| when find the lib and bin.

* style.

Co-authored-by: linquan <[email protected]>

build/copy-bin-lib.sh

@@ -20,15 +20,19 @@ function check_arch() {
 }
 
 function copy_lib() {
-  for target in $(find /usr -name "${1}*" | grep -v "stubs"); do
+  for target in $(find ${FIND_BASE} -name "${1}*" | grep -v "stubs"); do
     if [[ $(objdump -p ${target} 2>/dev/null | grep -o "SONAME") == "SONAME" ]]; then
       copy_directory ${target} "${NV_DIR}/lib$(check_arch ${target})"
     fi
   done
 }
 
 function copy_bin() {
-  for target in $(find /usr -name "${1}"); do
+  for target in $(find ${FIND_BASE} -name "${1}"); do
+    if [[ -L ${target} ]]; then
+      echo "${target} is symlink"
+      continue
+    fi
     copy_directory ${target} "${NV_DIR}/bin/"
   done
 }
@@ -80,4 +84,4 @@ done
   rm -rf libnvidia-ml.so
   rel_path=$(readlink -f libnvidia-ml.so.1)
   ln -s $(basename ${rel_path}) libnvidia-ml.so
-)
+)