C++中调用Pytorch模型

在C++中加载 #include <torch/script.h> 以使用 LibTorch（PyTorch 的 C++ API）时，需要进行一些设置，包括安装 LibTorch 库、配置编译器和链接器以找到库文件和头文件。

1. 下载和安装 LibTorch

首先，从 PyTorch 的官方页面下载 LibTorch。选择适合你的操作系统和编译器版本的包。下载链接可以在以下页面找到：

LibTorch 下载页面

选择 LibTorch 作为安装包类型。

2. 解压缩 LibTorch

下载完成后，解压缩文件。例如，在 Linux 或 macOS 上可以使用 tar 命令：

tar -xzf libtorch-shared-with-deps-latest.zip

在 Windows 上，使用解压工具解压到合适的目录。

3. 创建 C++ 项目

#include <torch/script.h> // 引入TorchScript头文件
#include <iostream>

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

4. 配置 CMake 构建系统

在项目根目录创建一个 CMakeLists.txt 文件，内容如下：

cmake_minimum_required(VERSION 3.10)
project(TorchExample)

set(CMAKE_CXX_STANDARD 14)

# 设置Torch的路径
set(Torch_DIR /path/to/libtorch/share/cmake/Torch)

find_package(Torch REQUIRED)

add_executable(TorchExample main.cpp)
target_link_libraries(TorchExample "${TORCH_LIBRARIES}")

set_property(TARGET TorchExample PROPERTY CXX_STANDARD 14)

5. 编译和运行项目

在项目根目录中创建一个 build 目录，并进入该目录：

mkdir build
cd build

然后运行 cmake 以生成构建文件：

cmake ..

接着运行 make 进行编译：

make

编译成功后，会生成一个名为 TorchExample 的可执行文件。运行它：

./TorchExample

得到以下输出：

tensor([[0.2542, 0.2022, 0.4594],
        [0.5168, 0.6440, 0.0237]])