@@ -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 "$@" |