libtorch在windows使用（c++项目中部署.pt）

原创

已于 2023-02-20 11:47:39 修改 · 1.9k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #pytorch #开发语言 #神经网络 #深度学习

于 2023-01-10 18:47:02 首次发布

本文围绕在C++中加载PyTorch模型展开，介绍了从已有model.pt开始加载的步骤，包括example-app.cpp、CMakeLists.txt内容及cmake指令。还阐述了遇到的问题，如版本对应、dll放置、cmake报错等，并给出解决办法，最后提及将代码嵌入项目的配置方法。

狗头

pytorch官网教程：Loading a TorchScript Model in C++ — PyTorch Tutorials 1.13.1+cu117 documentation

首先我已经有了个model.pt，就不需要做前面序列化为文件之类的操作，直接从在C++中加载开始

最后成功的版本：

example-app.cpp 内容：

#include <torch/script.h>
#include <torch/cuda.h>
int main(int argc, char* argv[]) {
    torch::jit::script::Module module;

    // auto  device = torch::Device(c10::DeviceType::CUDA, 0);
    //auto device = torch::kCUDA;

    try {
        std::cout << "before loading" << std::endl;
        module = torch::jit::load("D:\\project\\example-app\\model_withoutgpu.pt");
        // module.to(device);
        std::cout << "after loading" << std::endl;
    }