提交 365e80b5 编写于 作者: M Megvii Engine Team

fix(build): fix cu111 cudnn rpath

GitOrigin-RevId: 11e3227be1b3975714d5a563b3aa2cf0dc9176e4
上级 c89c6735
...@@ -32,10 +32,10 @@ function handle_copy_cuda_libs() { ...@@ -32,10 +32,10 @@ function handle_copy_cuda_libs() {
handle_strip ${TO_DIR}/libcuda_stub.so handle_strip ${TO_DIR}/libcuda_stub.so
cp /usr/local/cuda/lib64/libnvToolsExt.so.1 ${TO_DIR} cp /usr/local/cuda/lib64/libnvToolsExt.so.1 ${TO_DIR}
IFS=: read -a lib_name_array <<<"$CUDA_COPY_LIB_LIST" IFS=: read -a lib_name_array <<<"$CUDA_COPY_LIB_LIST"
append_rpath='$ORIGIN/.' append_rpath='$ORIGIN'
for lib_name in ${lib_name_array[@]};do for lib_name in ${lib_name_array[@]};do
echo "cuda copy detail: ${lib_name} to ${TO_DIR}" echo "cuda copy detail: ${lib_name} to ${TO_DIR}"
full_copy_so $lib_name ${TO_DIR} $lib_append_rpath full_copy_so $lib_name ${TO_DIR} $append_rpath
done done
fi fi
} }
......
#!/bin/bash -e #!/bin/bash -e
GET_PIP_URL='https://bootstrap.pypa.io/get-pip.py' GET_PIP_URL='https://bootstrap.pypa.io/get-pip.py'
GET_PIP_URL_35='https://bootstrap.pypa.io/3.5/get-pip.py' GET_PIP_URL_35='https://bootstrap.pypa.io/pip/3.5/get-pip.py'
SWIG_URL='https://downloads.sourceforge.net/project/swig/swig/swig-3.0.12/swig-3.0.12.tar.gz?use_mirror=autoselect' SWIG_URL='https://downloads.sourceforge.net/project/swig/swig/swig-3.0.12/swig-3.0.12.tar.gz?use_mirror=autoselect'
LLVM_URL='https://github.com/llvm-mirror/llvm/archive/release_60.tar.gz' LLVM_URL='https://github.com/llvm-mirror/llvm/archive/release_60.tar.gz'
CLANG_URL='https://github.com/llvm-mirror/clang/archive/release_60.tar.gz' CLANG_URL='https://github.com/llvm-mirror/clang/archive/release_60.tar.gz'
...@@ -67,7 +67,7 @@ popd >/dev/null ...@@ -67,7 +67,7 @@ popd >/dev/null
pushd /tmp >/dev/null pushd /tmp >/dev/null
curl -sSL https://github.com/NixOS/patchelf/archive/0.12.tar.gz | tar xz curl -sSL https://github.com/NixOS/patchelf/archive/0.12.tar.gz | tar xz
pushd /tmp/patchelf-0.12 >/dev/null pushd /tmp/patchelf-0.12 >/dev/null
sed -i '331s/32/64/' ./src/patchelf.cc sed -i '331s/32/256/' ./src/patchelf.cc
./bootstrap.sh && ./configure && make install-strip ./bootstrap.sh && ./configure && make install-strip
popd popd
rm -rf /tmp/patchelf-0.12 rm -rf /tmp/patchelf-0.12
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册