ubuntu18.04下libtorch的example的编译应用

搜索了好多方法，搞了很久，demo都编译不过，各种一堆错误，后来还是找到官网搞定的。

 https://pytorch.org/cppdocs/installing.html

1  下载libtorch

2 解压之后变成libtorch文件夹

3 新建example-app 文件夹和libtorch文件夹同级，如上图。

4 在example-app文件夹下面新建CMakeLists.txt 和example-app.cpp

CMakeLists.txt 文件内容：

cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(example-app)

find_package(Torch REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")

add_executable(example-app example-app.cpp)
target_link_libraries(example-app "${TORCH_LIBRARIES}")
set_property(TARGET example-app PROPERTY CXX_STANDARD 14)

 

example-app.cpp 文件内容：

#include <torch/torch.h>
#include <iostream>

int main() {
  torch::Tensor tensor = torch::rand({2, 3});
  std::cout << tensor << std::endl;
  return 0;
}

5 新建build文件夹，在build文件夹里面进行cmake准备，这里的绝对路径需要根据你的libtorch文件夹位置进行调整。

6 编译make。

7 执行应用程序。