From cb8ebb54c5eab50daf169703f738845f4d2026a4 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 8 Dec 2021 16:14:27 +0800 Subject: [PATCH] fix(mgb): whl use gcc8 GitOrigin-RevId: d8306839c282ecad80879edac66a95df551340e4 --- scripts/whl/manylinux2014/build_wheel_common.sh | 2 ++ 1 file changed, 2 insertions(+) diff --git a/scripts/whl/manylinux2014/build_wheel_common.sh b/scripts/whl/manylinux2014/build_wheel_common.sh index dfc40983..5cd80d3d 100755 --- a/scripts/whl/manylinux2014/build_wheel_common.sh +++ b/scripts/whl/manylinux2014/build_wheel_common.sh @@ -90,6 +90,7 @@ elif [ $SDK_NAME == "cu102_JetsonNano" ];then EXTRA_CMAKE_FLAG="-DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON -DMGE_CUDA_GENCODE=\"-gencode arch=compute_53,code=sm_53\" " elif [ $SDK_NAME == "cu111" ];then + BUILD_GCC8="ON" if [ ${machine} == "aarch64" ];then REQUIR_CUDA_VERSION="11010" REQUIR_CUDNN_VERSION="8.0.5" @@ -130,6 +131,7 @@ elif [ $SDK_NAME == "cu111" ];then fi elif [ $SDK_NAME == "cu112" ];then + BUILD_GCC8="ON" CUDA_COPY_LIB_LIST="\ ${CUDA_LIB_DIR}/libnvrtc.so.11.2:\ ${CUDA_LIB_DIR}/libcublasLt.so.11:\