Browse Source

fix(build): fix cu111 cudnn rpath

GitOrigin-RevId: 11e3227be1
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
365e80b599
2 changed files with 4 additions and 4 deletions
  1. +2
    -2
      scripts/whl/manylinux2014/do_build_common.sh
  2. +2
    -2
      scripts/whl/manylinux2014/init_image.sh

+ 2
- 2
scripts/whl/manylinux2014/do_build_common.sh View File

@@ -32,10 +32,10 @@ function handle_copy_cuda_libs() {
handle_strip ${TO_DIR}/libcuda_stub.so
cp /usr/local/cuda/lib64/libnvToolsExt.so.1 ${TO_DIR}
IFS=: read -a lib_name_array <<<"$CUDA_COPY_LIB_LIST"
append_rpath='$ORIGIN/.'
append_rpath='$ORIGIN'
for lib_name in ${lib_name_array[@]};do
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
fi
}


+ 2
- 2
scripts/whl/manylinux2014/init_image.sh View File

@@ -1,7 +1,7 @@
#!/bin/bash -e

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'
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'
@@ -67,7 +67,7 @@ popd >/dev/null
pushd /tmp >/dev/null
curl -sSL https://github.com/NixOS/patchelf/archive/0.12.tar.gz | tar xz
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
popd
rm -rf /tmp/patchelf-0.12


Loading…
Cancel
Save