Ubuntu cmakelist 添加链接libtorch
解决Ubuntu cmakelist 添加链接libtorch
C++工程中用到了libtorch库,在引用头文件时找不到torch.h
#include <torch/torch.h>
编译报错can’t find file xxx torch.h;
cmakelist中需要链接到libtorch库
set(CMAKE_PREFIX_PATH /home/xx/libtorch/share/cmake/Torch)
find_package(Torch REQUIRED)
include_directories(/home/xx/libtorch/include/torch/csrc/api/include/torch)
target_link_libraries(${project_name}
${TORCH_LIBRARIES})