@@ -1,3 +0,0 @@ | |||
/output/ | |||
/build_image.sh | |||
/build_wheel.sh |
@@ -1 +0,0 @@ | |||
/output/ |
@@ -1,15 +0,0 @@ | |||
FROM quay.io/pypa/manylinux2010_x86_64:2020-01-31-046f791 | |||
ENV UID=1024 \ | |||
PATH=${PATH}:/usr/local/cuda/bin \ | |||
LIBRARY_PATH=${LIBRARY_PATH}:/usr/local/cuda/lib64:/usr/local/cuda/lib64/stubs:/opt/cudnn/lib64:/opt/tensorrt/lib \ | |||
LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib64:/usr/local/cuda/lib64/stubs:/opt/cudnn/lib64:/opt/tensorrt/lib \ | |||
CPATH=${CPATH}:/usr/local/cuda/include:/opt/cudnn/include:/opt/tensorrt/include | |||
ARG platform | |||
COPY setup_mirror.sh . | |||
RUN ./setup_mirror.sh "$platform" | |||
ADD init_image.sh /tmp | |||
RUN /tmp/init_image.sh && rm -f /tmp/init_image.sh | |||
@@ -1,5 +0,0 @@ | |||
#!/bin/bash -e | |||
cd $(dirname $0) | |||
docker build -t env_manylinux2010:latest . |
@@ -1,103 +0,0 @@ | |||
#!/bin/bash -e | |||
CWD=$(dirname $0) | |||
BASEDIR=$(readlink -f ${CWD}/../../..) | |||
OUTPUTDIR=$(readlink -f ${CWD}/output) | |||
USERID=$(id -u) | |||
TMPFS_ARGS="--tmpfs /tmp:exec" | |||
BUILD_WHL_CPU_ONLY=${BUILD_WHL_CPU_ONLY} | |||
if [[ -z ${BUILD_WHL_CPU_ONLY} ]] | |||
then | |||
BUILD_WHL_CPU_ONLY="OFF" | |||
fi | |||
echo ${BASEDIR} | |||
pushd ${BASEDIR}/third_party >/dev/null | |||
./prepare.sh | |||
popd >/dev/null | |||
cd ${CWD} | |||
mkdir -p ${OUTPUTDIR} | |||
if [ ${BUILD_WHL_CPU_ONLY} = "OFF" ]; then | |||
if [[ -z ${CUDA_ROOT_DIR} ]]; then | |||
echo "Environment variable CUDA_ROOT_DIR not set." | |||
exit -1 | |||
fi | |||
if [[ -z ${CUDNN_ROOT_DIR} ]]; then | |||
echo "Environment variable CUDNN_ROOT_DIR not set." | |||
exit -1 | |||
fi | |||
if [[ -z ${TENSORRT_ROOT_DIR} ]]; then | |||
echo "Environment variable TENSORRT_ROOT_DIR not set." | |||
exit -1 | |||
fi | |||
## YOU SHOULD MODIFY CUDA VERSION AS BELOW WHEN UPGRADE | |||
REQUIR_CUDA_VERSION="10010" | |||
REQUIR_CUDNN_VERSION="7.6.3" | |||
REQUIR_TENSORRT_VERSION="6.0.1.5" | |||
CUDA_ROOT_DIR_=${CUDA_ROOT_DIR%*/} | |||
CUDNN_ROOT_DIR_=${CUDNN_ROOT_DIR%*/} | |||
TENSORRT_ROOT_DIR_=${TENSORRT_ROOT_DIR%*/} | |||
CUDA_VERSION_PATH=${CUDA_ROOT_DIR_}/include/cuda.h | |||
CUDNN_VERSION_PATH=${CUDNN_ROOT_DIR_}/include/cudnn.h | |||
TENSORRT_VERSION_PATH=${TENSORRT_ROOT_DIR_}/include/NvInferVersion.h | |||
if [ ! -e $CUDA_VERSION_PATH ] ; then | |||
echo file $CUDA_VERSION_PATH is not exist | |||
echo please check the Environment must use CUDA-10.1 NO.$REQUIR_CUDA_VERSION | |||
exit -1 | |||
fi | |||
if [ ! -e $CUDNN_VERSION_PATH ] ; then | |||
echo file $CUDNN_VERSION_PATH is not exist | |||
echo please check the Environment must use CUDNN-V$REQUIR_CUDNN_VERSION | |||
exit -1 | |||
fi | |||
if [ ! -e $TENSORRT_VERSION_PATH ] ; then | |||
echo file $TENSORRT_VERSION_PATH is not exist | |||
echo please check the Environment must use TensorRT-$REQUIR_TENSORRT_VERSION | |||
exit -1 | |||
fi | |||
CUDA_VERSION_CONTEXT=$(head -85 ${CUDA_VERSION_PATH}) | |||
CUDNN_VERSION_CONTEXT=$(head -62 ${CUDNN_VERSION_PATH}) | |||
TENSORRT_VERSION_CONTEXT=$(tail -12 ${TENSORRT_VERSION_PATH}) | |||
CUDA_API_VERSION=$(echo $CUDA_VERSION_CONTEXT | grep -Eo "define __CUDA_API_VERSION * +([0-9]+)") | |||
CUDA_VERSION=${CUDA_API_VERSION:0-5} | |||
echo CUDA_VERSION:$CUDA_VERSION | |||
CUDNN_VERSION_MAJOR=$(echo $CUDNN_VERSION_CONTEXT | grep -Eo "define CUDNN_MAJOR * +([0-9]+)") | |||
CUDNN_VERSION_MINOR=$(echo $CUDNN_VERSION_CONTEXT | grep -Eo "define CUDNN_MINOR * +([0-9]+)") | |||
CUDNN_VERSION_PATCH=$(echo $CUDNN_VERSION_CONTEXT | grep -Eo "define CUDNN_PATCHLEVEL * +([0-9]+)") | |||
CUDNN_VERSION=${CUDNN_VERSION_MAJOR:0-1}.${CUDNN_VERSION_MINOR:0-1}.${CUDNN_VERSION_PATCH:0-1} | |||
echo CUDNN_VERSION:$CUDNN_VERSION | |||
TENSORRT_VERSION_MAJOR=$(echo $TENSORRT_VERSION_CONTEXT | grep -Eo "NV_TENSORRT_MAJOR * +([0-9]+)") | |||
TENSORRT_VERSION_MINOR=$(echo $TENSORRT_VERSION_CONTEXT | grep -Eo "NV_TENSORRT_MINOR * +([0-9]+)") | |||
TENSORRT_VERSION_PATCH=$(echo $TENSORRT_VERSION_CONTEXT | grep -Eo "NV_TENSORRT_PATCH * +([0-9]+)") | |||
TENSORRT_VERSION_BUILD=$(echo $TENSORRT_VERSION_CONTEXT | grep -Eo "NV_TENSORRT_BUILD * +([0-9]+)") | |||
TENSORRT_VERSION=${TENSORRT_VERSION_MAJOR:0-1}.${TENSORRT_VERSION_MINOR:0-1}.${TENSORRT_VERSION_PATCH:0-1}.${TENSORRT_VERSION_BUILD:0-1} | |||
echo TENSORRT_VERSION:$TENSORRT_VERSION | |||
if [ $CUDA_VERSION != $REQUIR_CUDA_VERSION ] ; then | |||
echo please check the Environment must use CUDA-10.1 NO.$REQUIR_CUDA_VERSION | |||
exit -1 | |||
fi | |||
if [ $CUDNN_VERSION != $REQUIR_CUDNN_VERSION ] ; then | |||
echo please check the Environment must use CUDNN-V$REQUIR_CUDNN_VERSION | |||
exit -1 | |||
fi | |||
if [ $TENSORRT_VERSION != $REQUIR_TENSORRT_VERSION ] ; then | |||
echo please check the Environment must use TENSORRT-$REQUIR_TENSORRT_VERSION | |||
exit -1 | |||
fi | |||
fi | |||
docker run -it --rm $TMPFS_ARGS -e UID=${USERID} -e LOCAL_VERSION=${LOCAL_VERSION} -e BUILD_WHL_CPU_ONLY=${BUILD_WHL_CPU_ONLY} -e ALL_PYTHON="${ALL_PYTHON}" -v ${CUDA_ROOT_DIR}:/usr/local/cuda -v ${CUDNN_ROOT_DIR}:/opt/cudnn -v ${TENSORRT_ROOT_DIR}:/opt/tensorrt -v ${BASEDIR}:/home/code -v ${OUTPUTDIR}:/home/output:rw env_manylinux2010:latest /home/code/scripts/whl/manylinux2010/do_build.sh |
@@ -1,117 +0,0 @@ | |||
#!/bin/bash -e | |||
ALL_PYTHON=${ALL_PYTHON} | |||
if [[ -z ${ALL_PYTHON} ]] | |||
then | |||
ALL_PYTHON="35m 36m 37m 38" | |||
fi | |||
BUILD_WHL_CPU_ONLY=${BUILD_WHL_CPU_ONLY} | |||
if [[ -z ${BUILD_WHL_CPU_ONLY} ]] | |||
then | |||
BUILD_WHL_CPU_ONLY="OFF" | |||
fi | |||
SRC_DIR=$(readlink -f "`dirname $0`/../../../") | |||
BUILD_DIR=${SRC_DIR}/build_dir/host/MGE_WITH_CUDA_OFF/MGE_INFERENCE_ONLY_OFF/Release/build/ | |||
if [ ${BUILD_WHL_CPU_ONLY} = "OFF" ]; then | |||
BUILD_DIR=${SRC_DIR}/build_dir/host/MGE_WITH_CUDA_ON/MGE_INFERENCE_ONLY_OFF/Release/build/ | |||
fi | |||
NEW_LIB_PATH=core/lib | |||
function handle_strip() { | |||
echo "now handle strip $1" | |||
objcopy --only-keep-debug $1 $1.dbg | |||
strip -s $1 | |||
objcopy --add-gnu-debuglink=$1.dbg $1 | |||
rm $1.dbg | |||
} | |||
function patch_elf_depend_lib() { | |||
echo "handle common depend lib" | |||
LIBS_DIR=${BUILD_DIR}/staging/megengine/core/lib | |||
mkdir -p ${LIBS_DIR} | |||
cp /usr/lib64/libatomic.so.1 ${LIBS_DIR} | |||
patchelf --remove-rpath ${BUILD_DIR}/staging/megengine/core/_imperative_rt.so | |||
patchelf --force-rpath --set-rpath '$ORIGIN/lib' ${BUILD_DIR}/staging/megengine/core/_imperative_rt.so | |||
cp ${BUILD_DIR}/src/libmegengine_export.so ${LIBS_DIR} | |||
patchelf --remove-rpath ${LIBS_DIR}/libmegengine_export.so | |||
patchelf --force-rpath --set-rpath '$ORIGIN/.' ${LIBS_DIR}/libmegengine_export.so | |||
if [ ${BUILD_WHL_CPU_ONLY} = "OFF" ]; then | |||
echo "handle cuda lib" | |||
CUDA_VER=10.1 | |||
cp ${BUILD_DIR}/dnn/cuda-stub/libcuda.so ${LIBS_DIR} | |||
cp /usr/local/cuda/lib64/libnvrtc.so.${CUDA_VER} ${LIBS_DIR} | |||
cp /usr/local/cuda/lib64/libnvToolsExt.so.1 ${LIBS_DIR} | |||
fi | |||
} | |||
for ver in ${ALL_PYTHON} | |||
do | |||
USE_AUDITWHEEL="ON" | |||
python_ver=${ver:0:2} | |||
MAJOR=${python_ver:0:1} | |||
MINOR=${ver:1} | |||
PYTHON_DIR=/opt/python/cp${python_ver}-cp${ver}/ | |||
export EXTRA_CMAKE_ARGS="${EXTRA_CMAKE_ARGS} -DCMAKE_BUILD_TYPE=RelWithDebInfo" | |||
export EXTRA_CMAKE_ARGS="${EXTRA_CMAKE_ARGS} -DCMAKE_PREFIX_PATH=${PYTHON_DIR}" | |||
export EXTRA_CMAKE_ARGS="${EXTRA_CMAKE_ARGS} -DPYTHON_EXECUTABLE=${PYTHON_DIR}/bin/python3" | |||
export EXTRA_CMAKE_ARGS="${EXTRA_CMAKE_ARGS} -DPYTHON_LIBRARY=${PYTHON_DIR}lib/" | |||
export EXTRA_CMAKE_ARGS="${EXTRA_CMAKE_ARGS} -DPYTHON_INCLUDE_DIR=${PYTHON_DIR}include/python${MAJOR}.${MINOR}" | |||
export EXTRA_CMAKE_ARGS="${EXTRA_CMAKE_ARGS} -DMGE_WITH_ATLAS=ON" | |||
if [ ${BUILD_WHL_CPU_ONLY} = "OFF" ]; then | |||
${SRC_DIR}/scripts/cmake-build/host_build.sh -c -t -r | |||
else | |||
${SRC_DIR}/scripts/cmake-build/host_build.sh -t -r | |||
fi | |||
cd ${BUILD_DIR} | |||
rm -rf staging | |||
mkdir -p staging | |||
cp -a imperative/python/{megengine,setup.py,requires.txt,requires-style.txt,requires-test.txt} staging/ | |||
if [ ${BUILD_WHL_CPU_ONLY} = "OFF" ]; then | |||
cd ${BUILD_DIR}/dnn/cuda-stub/ | |||
strip -s libcuda.so | |||
ln -sf libcuda.so libcuda.so.1 | |||
fi | |||
handle_strip ${BUILD_DIR}/src/libmegengine_export.so | |||
cd ${BUILD_DIR}/staging/megengine/core | |||
handle_strip _imperative_rt.so | |||
mkdir -p lib/ucx | |||
if [ ${BUILD_WHL_CPU_ONLY} = "OFF" ]; then | |||
cp -L /usr/local/cuda/lib*/libnvrtc-builtins.so lib | |||
fi | |||
if [ ${USE_AUDITWHEEL} = "OFF" ]; then | |||
patch_elf_depend_lib | |||
fi | |||
cd ${BUILD_DIR}/staging/ | |||
${PYTHON_DIR}/bin/python setup.py bdist_wheel | |||
cd /home/output | |||
if [ ${USE_AUDITWHEEL} = "ON" ]; then | |||
LD_LIBRARY_PATH=${BUILD_DIR}/dnn/cuda-stub:$LD_LIBRARY_PATH auditwheel repair -L ${NEW_LIB_PATH} ${BUILD_DIR}/staging/dist/Meg*.whl | |||
else | |||
mkdir -p ${SRC_DIR}/scripts/whl/manylinux2010/output/wheelhouse | |||
cd ${BUILD_DIR}/staging/dist/ | |||
org_whl_name=`ls Meg*${ver}*.whl` | |||
compat_whl_name=`echo ${org_whl_name} | sed 's/linux/manylinux2010/'` | |||
echo "org whl name: ${org_whl_name}" | |||
echo "comapt whl name: ${compat_whl_name}" | |||
mv ${org_whl_name} ${SRC_DIR}/scripts/whl/manylinux2010/output/wheelhouse/${compat_whl_name} | |||
cd /home/output | |||
fi | |||
chown -R ${UID}.${UID} . | |||
# compat for root-less docker env to remove output at host side | |||
chmod -R 777 . | |||
done |
@@ -1,100 +0,0 @@ | |||
#!/bin/bash -e | |||
GET_PIP_URL='https://bootstrap.pypa.io/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' | |||
yum erase -y cmake cmake28 | |||
yum install -y python34-pip pcre-devel devtoolset-8-libatomic-devel.x86_64 | |||
pip3 install --no-cache-dir --only-binary :all: -U pip==19.1 | |||
pip3 install --no-cache-dir --only-binary :all: cmake==3.16.3 | |||
for ver in 35m 36m 37m 38 | |||
do | |||
python_ver=${ver:0:2} | |||
curl ${GET_PIP_URL} | /opt/python/cp${python_ver}-cp${ver}/bin/python - \ | |||
--no-cache-dir --only-binary :all: | |||
/opt/python/cp${python_ver}-cp${ver}/bin/pip install \ | |||
--no-cache-dir --only-binary :all: numpy==1.18.1 setuptools==46.1.3 | |||
done | |||
pushd /home >/dev/null | |||
echo "Install swig" | |||
curl -sSL ${SWIG_URL} | tar xz | |||
pushd swig-3.0.12 >/dev/null | |||
mkdir build | |||
pushd build >/dev/null | |||
../configure | |||
make -j$(nproc) | |||
make install | |||
popd >/dev/null | |||
popd >/dev/null | |||
rm -rf swig-3.0.12 | |||
echo "Install llvm" | |||
curl -sSL ${LLVM_URL} | tar xz | |||
pushd llvm-release_60 >/dev/null | |||
mkdir build | |||
pushd build >/dev/null | |||
cmake .. -DCMAKE_PREFIX_PATH=/opt/python/cp36-cp36m/ \ | |||
-DCMAKE_BUILD_TYPE=Release | |||
make -j$(nproc) | |||
make install | |||
popd >/dev/null | |||
popd >/dev/null | |||
rm -rf llvm-release_60 | |||
echo "Install clang" | |||
curl -sSL ${CLANG_URL} | tar xz | |||
pushd clang-release_60 >/dev/null | |||
mkdir build | |||
pushd build >/dev/null | |||
cmake .. -DCMAKE_PREFIX_PATH=/opt/python/cp36-cp36m/ \ | |||
-DCMAKE_BUILD_TYPE=Release | |||
make -j$(nproc) | |||
make install | |||
popd >/dev/null | |||
popd >/dev/null | |||
rm -rf clang-release_60 | |||
popd >/dev/null | |||
pushd /tmp >/dev/null | |||
curl -sSL https://github.com/NixOS/patchelf/archive/0.10.tar.gz | tar xz | |||
pushd /tmp/patchelf-0.10 >/dev/null | |||
patch -p1 <<'EOF' | |||
diff --git a/src/patchelf.cc b/src/patchelf.cc | |||
index 0b4965a..7aae7a4 100644 | |||
--- a/src/patchelf.cc | |||
+++ b/src/patchelf.cc | |||
@@ -1074,13 +1074,6 @@ void ElfFile<ElfFileParamNames>::modifySoname(sonameMode op, const std::string & | |||
return; | |||
} | |||
- /* Zero out the previous SONAME */ | |||
- unsigned int sonameSize = 0; | |||
- if (soname) { | |||
- sonameSize = strlen(soname); | |||
- memset(soname, 'X', sonameSize); | |||
- } | |||
- | |||
debug("new SONAME is '%s'\n", newSoname.c_str()); | |||
/* Grow the .dynstr section to make room for the new SONAME. */ | |||
@@ -1264,7 +1257,6 @@ void ElfFile<ElfFileParamNames>::modifyRPath(RPathOp op, | |||
unsigned int rpathSize = 0; | |||
if (rpath) { | |||
rpathSize = strlen(rpath); | |||
- memset(rpath, 'X', rpathSize); | |||
} | |||
debug("new rpath is '%s'\n", newRPath.c_str()); | |||
EOF | |||
./bootstrap.sh && ./configure && make install-strip | |||
popd | |||
rm -rf /tmp/patchelf-0.10 | |||
popd | |||
yum clean all |
@@ -1,65 +0,0 @@ | |||
#!/bin/bash | |||
set -e | |||
function set_tuna_yum_mirror() { | |||
cp /etc/yum.repos.d/CentOS-Base.repo /etc/yum.repos.d/CentOS-Base.repo.bak | |||
local repo=/etc/yum.repos.d/CentOS-Base.repo | |||
local plugin=/etc/yum/pluginconf.d/fastestmirror.conf | |||
sed -i "s/mirrorlist=/#mirrorlist=/g" $repo | |||
sed -i "s/#baseurl/baseurl/g" $repo | |||
sed -i "s/mirror.centos.org/mirrors.tuna.tsinghua.edu.cn/g" $repo | |||
sed -i "s/http/https/g" $repo | |||
sed -i "s/enabled=1/enabled=0/g" $plugin | |||
yum clean all | |||
# Build on brainpp unable to pull epel reo metadata so disable this | |||
# https://unix.stackexchange.com/questions/148144/unable-to-pull-epel-repository-metadata | |||
yum --disablerepo="epel" update nss | |||
yum makecache | |||
} | |||
function set_epel() { | |||
mv /etc/yum.repos.d/epel.repo /etc/yum.repos.d/epel.repo.backup | |||
mv /etc/yum.repos.d/epel-testing.repo /etc/yum.repos.d/epel-testing.repo.backup | |||
curl -o /etc/yum.repos.d/epel.repo http://mirrors.aliyun.com/repo/epel-6.repo | |||
} | |||
function set_yum_mirror() { | |||
mv /etc/yum.repos.d/CentOS-Base.repo /etc/yum.repos.d/CentOS-Base.repo.backup | |||
curl -o /etc/yum.repos.d/CentOS-Base.repo https://mirrors.aliyun.com/repo/Centos-6.repo | |||
yum makecache | |||
} | |||
function set_pip_mirror() { | |||
cat > /etc/pip.conf <<EOF | |||
[global] | |||
timeout = 180 | |||
index-url = https://mirrors.aliyun.com/pypi/simple | |||
extra-index-url = | |||
http://mirrors.i.brainpp.cn/pypi/simple/ | |||
http://pypi.i.brainpp.cn/brain/dev/+simple | |||
https://pypi.tuna.tsinghua.edu.cn/simple | |||
trusted-host = | |||
mirrors.i.brainpp.cn | |||
pypi.i.brainpp.cn | |||
pypi.tuna.tsinghua.edu.cn | |||
mirrors.aliyun.com | |||
EOF | |||
} | |||
function main() { | |||
local platform=$1 | |||
case $platform in | |||
brainpp) | |||
set_epel | |||
set_yum_mirror | |||
set_pip_mirror | |||
;; | |||
*) | |||
echo "No setup required" | |||
;; | |||
esac | |||
} | |||
main "$@" |