Linux Pytorch C++

0 安装Linux环境

apt-get install cmake
apt-get install libnss3

1，下载Pytorch C++编译环境

https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip

并解压

 

2 安装vscode

   下载地址：https://vscode.cdn.azure.cn/stable/7f3ce96ff4729c91352ae6def877e59c561f4850/code-stable-1539735949.tar.gz

   启动

./code

3 pytorch c++编写

创建一个工程，目录结构：

test/
  a.cpp
  CMakeLists.txt

CMakeLists.txt，代码

cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(custom_ops)

find_package(Torch REQUIRED)

add_executable(a a.cpp)
target_link_libraries(a "${TORCH_LIBRARIES}")
set_property(TARGET a PROPERTY CXX_STANDARD 11)

a.cpp，代码：

#include <torch/script.h> // One-stop header.
#include <ATen/ATen.h>

#include <iostream>
#include <memory>

using namespace std;
using namespace at;
 
int main(int argc, const char* argv[]) {

  at::Tensor a = at::ones({2, 2}, at::kInt);
  std::cout << a << "\n";

  std::cout << "ok\n";
  getchar();

  return 1;
}

4，编译并执行。

4.1 编译

在工程下创建lib,工程目录为:

test/
  a.cpp
  CMakeLists.txt
  lib --目录

cd lib
cmake -DCMAKE_PREFIX_PATH=/path/to/libtorch(pytorch C++ 解压的绝对路径) ..
make

4.2 执行:

根据CMakeLists.txt，会生成a的执行文件。

./a