Browse Source

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

GitOrigin-RevId: 607af27256
release-1.11.1
Megvii Engine Team 2 years ago
parent
commit
f0f6f5fe5c
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      scripts/whl/manylinux2014/build_wheel_common.sh

+ 3
- 0
scripts/whl/manylinux2014/build_wheel_common.sh View File

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


Loading…
Cancel
Save