|
|
|
@@ -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" |
|
|
|
|
Megvii Engine Team
2 years ago