2021-07-23

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})