From c6ff878d876ce292c227bb8a9dc13d797ac92b1e Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 19 Aug 2022 16:58:03 +0800 Subject: [PATCH] feat(mgb): add cu114 wheel GitOrigin-RevId: 013c1cfa503af2e75a7f3f4f89c770bbc5e150c3 --- .../whl/manylinux2014/build_wheel_common.sh | 32 ++++++++++++++++++- 1 file changed, 31 insertions(+), 1 deletion(-) diff --git a/scripts/whl/manylinux2014/build_wheel_common.sh b/scripts/whl/manylinux2014/build_wheel_common.sh index 2b2315461..edbb833f8 100755 --- a/scripts/whl/manylinux2014/build_wheel_common.sh +++ b/scripts/whl/manylinux2014/build_wheel_common.sh @@ -12,7 +12,7 @@ CUDA_LIB_DIR="/usr/local/cuda/lib64/" TensorRT_LIB_DIR="/opt/tensorrt/lib/" SDK_NAME="unknown" -x86_64_support_version="cu101 cu111 cu112 cpu cu111_cudnn821_tensorRT825" +x86_64_support_version="cu101 cu111 cu112 cpu cu111_cudnn821_tensorRT825 cu114" aarch64_support_version="cu102_JetsonNano cu111 cpu" if [[ -z ${IN_CI} ]] then @@ -193,6 +193,36 @@ elif [ $SDK_NAME == "cu112" ];then REQUIR_TENSORRT_VERSION="7.2.2.3" REQUIR_CUBLAS_VERSION="11.3.1.68" + +elif [ $SDK_NAME == "cu114" ];then + BUILD_GCC8="ON" + REQUIR_CUDA_VERSION="11040" + REQUIR_CUDNN_VERSION="8.2.1" + REQUIR_TENSORRT_VERSION="7.2.2.3" + REQUIR_CUBLAS_VERSION="11.6.5.2" + + + CUDA_COPY_LIB_LIST="\ + ${CUDA_LIB_DIR}/libnvrtc.so.11.2:\ + ${CUDA_LIB_DIR}/libcublasLt.so.11:\ + ${CUDA_LIB_DIR}/libcublas.so.11:\ + ${CUDNN_LIB_DIR}/libcudnn_adv_infer.so.8:\ + ${CUDNN_LIB_DIR}/libcudnn_adv_train.so.8:\ + ${CUDNN_LIB_DIR}/libcudnn_cnn_infer.so.8:\ + ${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" + + EXTRA_CMAKE_FLAG=" -DMGE_WITH_CUDNN_SHARED=ON -DMGE_WITH_CUBLAS_SHARED=ON \ + -DMGE_CUDA_GENCODE=\"-gencode arch=compute_61,code=sm_61 \ + -gencode arch=compute_70,code=sm_70 \ + -gencode arch=compute_75,code=sm_75 \ + -gencode arch=compute_80,code=sm_80 \ + -gencode arch=compute_86,code=sm_86 \ + -gencode arch=compute_86,code=compute_86\" " + + elif [ $SDK_NAME == "cpu" ];then echo "use $SDK_NAME without cuda support" BUILD_WHL_CPU_ONLY="ON" -- GitLab