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:\