diff --git a/scripts/whl/manylinux2014/build_wheel_common.sh b/scripts/whl/manylinux2014/build_wheel_common.sh index 2b2315461a3a3123d01faa99124817facac1929e..edbb833f8149851ee674056a6c3b2d715026eb9e 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"