mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-10 02:50:19 +08:00
64 lines
2.8 KiB
CMake
64 lines
2.8 KiB
CMake
CMAKE_MINIMUM_REQUIRED (VERSION 3.16)
|
|
|
|
set(WITH_GPU @WITH_GPU@)
|
|
set(ENABLE_ORT_BACKEND @ENABLE_ORT_BACKEND@)
|
|
set(ENABLE_TRT_BACKEND @ENABLE_TRT_BACKEND&)
|
|
set(ENABLE_PADDLE_FRONTEND @ENABLE_PADDLE_FRONTEND@)
|
|
set(ENABLE_VISION @ENABLE_VISION@)
|
|
set(ENABLE_OPENCV_CUDA @ENABLE_OPENCV_CUDA@)
|
|
|
|
set(FASTDEPLOY_LIBS "")
|
|
set(FASTDEPLOY_INCS "")
|
|
list(APPEND FASTDEPLOY_INCS ${CMAKE_CURRENT_LIST_DIR}/include)
|
|
|
|
if(NOT CMAKE_CXX_STANDARD)
|
|
set(CMAKE_CXX_STANDARD 11)
|
|
endif()
|
|
|
|
find_library(FDLIB fastdeploy ${CMAKE_CURRENT_LIST_DIR}/lib)
|
|
list(APPEND FASTDEPLOY_LIBS ${FDLIB})
|
|
if(ENABLE_ORT_BACKEND)
|
|
find_library(ORT_LIB onnxruntime ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/onnxruntime/lib)
|
|
list(APPEND FASTDEPLOY_LIBS ${ORT_LIB})
|
|
endif()
|
|
|
|
if(WITH_GPU)
|
|
if (NOT CUDA_DIRECTORY)
|
|
message(FATAL_ERROR "[FastDeploy] Please define CUDA_DIRECTORY, e.g -DCUDA_DIRECTORY=/usr/local/cuda")
|
|
endif()
|
|
find_library(CUDA_LIB cudart ${CUDA_DIRECTORY}/lib64)
|
|
list(APPEND FASTDEPLOY_LIBS ${CUDA_LIB})
|
|
|
|
if (ENABLE_TRT_BACKEND)
|
|
if (NOT TRT_DIRECTORY)
|
|
message(FATAL_ERROR "[FastDeploy] Please define TRT_DIRECTORY, e.g -DTRT_DIRECTORY=/usr/downloads/TensorRT-8.4.1.0")
|
|
endif()
|
|
find_library(TRT_INFER_LIB nvinfer ${TRT_DIRECTORY}/lib)
|
|
find_library(TRT_ONNX_LIB nvonnxparser ${TRT_DIRECTORY}/lib)
|
|
find_library(TRT_CAFFE_LIB nvcaffe_parser ${TRT_DIRECTORY}/lib)
|
|
find_library(TRT_PLUGIN_LIB nvinfer_plugin ${TRT_DIRECTORY}/lib)
|
|
list(APPEND FASTDEPLOY_LIBS ${TRT_INFER_LIB} ${TRT_ONNX_LIB} ${TRT_CAFFE_LIB} ${TRT_PLUGIN_LIB})
|
|
endif()
|
|
endif()
|
|
|
|
if(ENABLE_VISION)
|
|
find_library(OPENCV_CORE_LIB opencv_core ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/opencv/lib)
|
|
find_library(OPENCV_HIGHGUI_LIB opencv_highgui ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/opencv/lib)
|
|
find_library(OPENCV_IMGPROC_LIB opencv_imgproc ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/opencv/lib)
|
|
find_library(OPENCV_IMGCODESC_LIB opencv_imgcodecs ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/opencv/lib)
|
|
list(APPEND FASTDEPLOY_LIBS ${OPENCV_CORE_LIB} ${OPENCV_HIGHGUI_LIB} ${OPENCV_IMGPROC_LIB} ${OPENCV_IMGCODESC_LIB})
|
|
list(APPEND FASTDEPLOY_INCS ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/opencv/include)
|
|
|
|
if(ENABLE_OPENCV_CUDA)
|
|
find_library(OPENCV_CUDAARITHM_LIB opencv_core ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/opencv/lib)
|
|
find_library(OPENCV_CUDAIMGPROC_LIB opencv_core ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/opencv/lib)
|
|
find_library(OPENCV_CUDAWARPING_LIB opencv_core ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/opencv/lib)
|
|
list(APPEND FASTDEPLOY_LIBS ${OPENCV_CUDAARITHM_LIB} ${OPENCV_CUDAIMGPROC_LIB} ${OPENCV_CUDAWARPING_LIB})
|
|
endif()
|
|
endif()
|
|
|
|
if(ENABLE_PADDLE_FRONTEND)
|
|
find_library(PADDLE2ONNX_LIB paddle2onnx ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/paddle2onnx/lib)
|
|
list(APPEND FASTDEPLOY_LIBS ${PADDLE2ONNX_LIB})
|
|
endif()
|