Linux系统使用Cmake编译Libtorch和OpenCV

CMakeLists.txt的设置如下即可，注意将Torch_DIR的路径更换为自己的libtorch存放的路径．

cmake_minimum_required(VERSION 3.5)

project(libtorch-yolov5)

set(CMAKE_CXX_STANDARD 14)

set(CMAKE_CXX_STANDARD_REQUIRED ON)

set(CMAKE_CXX_EXTENSIONS OFF)

set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall")

find_package(OpenCV REQUIRED)

set(Torch_DIR /home/think/libtorch)

find_package(Torch PATHS ${Torch_DIR} NO_DEFAULT REQUIRED)

add_executable(${CMAKE_PROJECT_NAME} ${SOURCE_FILES})

target_link_libraries (

${CMAKE_PROJECT_NAME}

${OpenCV_LIBS}

${TORCH_LIBRARIES}

)