From 83fb1622a45d0f4ef50453779eabfe044c6c83b0 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 8 Jul 2022 18:53:06 +0800 Subject: [PATCH] build(wheel): copy libcupti.so GitOrigin-RevId: 8ff807012dc6ae614909fe9ae54cd4faad66e60b --- scripts/whl/manylinux2014/build_wheel_common.sh | 25 +++++++++---------------- 1 file changed, 9 insertions(+), 16 deletions(-) diff --git a/scripts/whl/manylinux2014/build_wheel_common.sh b/scripts/whl/manylinux2014/build_wheel_common.sh index 2ea03aaf..6b182fd4 100755 --- a/scripts/whl/manylinux2014/build_wheel_common.sh +++ b/scripts/whl/manylinux2014/build_wheel_common.sh @@ -60,7 +60,9 @@ fi echo "Build with ${SDK_NAME}" if [ $SDK_NAME == "cu101" ];then - CUDA_COPY_LIB_LIST="${CUDA_LIB_DIR}/libnvrtc.so.10.1" + CUDA_COPY_LIB_LIST="\ + ${CUDA_LIB_DIR}/libnvrtc.so.10.1:\ + ${CUDA_LIB_DIR}/libcupti.so.10.1" EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=OFF -DMGE_WITH_CUBLAS_SHARED=OFF" BUILD_GCC8="ON" REQUIR_CUDA_VERSION="10010" @@ -85,13 +87,8 @@ elif [ $SDK_NAME == "cu102_JetsonNano" ];then ${CUDNN_LIB_DIR}/libcudnn_cnn_train.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_ops_infer.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_ops_train.so.8:\ - ${CUDNN_LIB_DIR}/libcudnn.so.8" - - if [ ${machine} == "aarch64" ];then - CUDA_COPY_LIB_LIST="\ - ${CUDA_LIB_DIR}/libcupti.so.10.2:\ - ${CUDA_COPY_LIB_LIST}" - fi + ${CUDNN_LIB_DIR}/libcudnn.so.8:\ + ${CUDA_LIB_DIR}/libcupti.so.10.2" EXTRA_CMAKE_FLAG="-DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON -DMGE_CUDA_GENCODE=\"-gencode arch=compute_53,code=sm_53\" " @@ -122,13 +119,8 @@ elif [ $SDK_NAME == "cu111" ];then ${CUDNN_LIB_DIR}/libcudnn_cnn_train.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_ops_infer.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_ops_train.so.8:\ - ${CUDNN_LIB_DIR}/libcudnn.so.8" - - if [ ${machine} == "aarch64" ];then - CUDA_COPY_LIB_LIST="\ - ${CUDA_LIB_DIR}/libcupti.so.11.1:\ - ${CUDA_COPY_LIB_LIST}" - fi + ${CUDNN_LIB_DIR}/libcudnn.so.8:\ + ${CUDNN_LIB_DIR}/libcupti.so.11.1" if [ ${IN_CI} = "true" ] && [ ${machine} == "aarch64" ]; then EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON -DMGE_CUDA_GENCODE=\"-gencode arch=compute_75,code=sm_75\" " @@ -154,7 +146,8 @@ elif [ $SDK_NAME == "cu112" ];then ${CUDNN_LIB_DIR}/libcudnn_cnn_train.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_ops_infer.so.8:\ ${CUDNN_LIB_DIR}/libcudnn_ops_train.so.8:\ - ${CUDNN_LIB_DIR}/libcudnn.so.8" + ${CUDNN_LIB_DIR}/libcudnn.so.8:\ + ${CUDNN_LIB_DIR}/libcupti.so.11.2" EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON \ -DMGE_CUDA_GENCODE=\"-gencode arch=compute_61,code=sm_61 \