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

build(mge/whl): package libnvrtc-builtins.so.11.x

GitOrigin-RevId: 607af2725629faf05b700b6cd15dd98d8c2ea703
上级 fb9d73b2
......@@ -113,6 +113,7 @@ elif [ $SDK_NAME == "cu111" ];then
CUDA_COPY_LIB_LIST="\
${CUDA_LIB_DIR}/libnvrtc.so.11.1:\
${CUDA_LIB_DIR}/libnvrtc-builtins.so.11.1:\
${CUDA_LIB_DIR}/libcublasLt.so.11:\
${CUDA_LIB_DIR}/libcublas.so.11:\
${CUDNN_LIB_DIR}/libcudnn_adv_infer.so.8:\
......@@ -145,6 +146,7 @@ elif [ $SDK_NAME == "cu111_cudnn821_tensorRT825" ];then
CUDA_COPY_LIB_LIST="\
${CUDA_LIB_DIR}/libnvrtc.so.11.1:\
${CUDA_LIB_DIR}/libnvrtc-builtins.so.11.1:\
${CUDA_LIB_DIR}/libcublasLt.so.11:\
${CUDA_LIB_DIR}/libcublas.so.11:\
${CUDNN_LIB_DIR}/libcudnn_adv_infer.so.8:\
......@@ -170,6 +172,7 @@ elif [ $SDK_NAME == "cu112" ];then
BUILD_GCC8="ON"
CUDA_COPY_LIB_LIST="\
${CUDA_LIB_DIR}/libnvrtc.so.11.2:\
${CUDA_LIB_DIR}/libnvrtc-builtins.so.11.2:\
${CUDA_LIB_DIR}/libcublasLt.so.11:\
${CUDA_LIB_DIR}/libcublas.so.11:\
${CUDNN_LIB_DIR}/libcudnn_adv_infer.so.8:\
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册