提交 cb8ebb54 编写于 作者: M Megvii Engine Team 提交者: XindaH

fix(mgb): whl use gcc8

GitOrigin-RevId: d8306839c282ecad80879edac66a95df551340e4
上级 315b908c
......@@ -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:\
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册