Hi,
Here is a sample for your reference:
cmake_minimum_required(VERSION 2.8)
project(TensorRT-sample)
set(CUDA_USE_STATIC_CUDA_RUNTIME OFF)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
set(
CUDA_NVCC_FLAGS
${CUDA_NVCC_FLAGS};
-O3
-gencode arch=compute_53,code=sm_53
-gencode arch=compute_62,code=sm_62
)
find_package(CUDA)
find_package(OpenCV REQUIRED)
include_directories(${OpenCV_INCLUDE_DIRS} /usr/local/cuda/include /usr/include/aarch64-linux-gnu)
link_directories(/usr/lib/aarch64-linux-gnu)
add_executable(sample_fasterRCNN sampleFasterRCNN.cpp)
target_link_libraries(sample_fasterRCNN nvcaffe_parser nvinfer_plugin nvinfer ${OpenCV_LIBS})
You can find some information in our jetson_inference sample:
And OpenCV tutorial:
Thanks.