时间戳

首页 > 解决方案 > 在 C++ 中加载 pytorch 模型，libtorch.dll 的问题

c++ - 在 C++ 中加载 pytorch 模型，libtorch.dll 的问题

问题描述

我正在尝试将使用 pytorch 训练的神经网络模型加载到 C++ 程序中。有一个教程如何做到这一点，但不能让它工作。

控制台出现，然后我得到，

“代码执行无法继续，因为找不到对象 xxx.dll”。

有时是 c10.dll，有时是 torch.dll 或 caffe2.dll。

我检查了我的 C/C++ 和 Linker 属性 50 次。
我使用 libtorch 的调试和发布版本对其进行了检查。
我在 Debug x86、Debug x64、Release x86、Release x64 上运行它。
我手动将这些文件添加到 Debug 文件夹（.exe 文件所在的位置）
，但没有一个起作用。

这是我的代码

#include "pch.h"
#include <torch/script.h>
#include <memory>
#include <iostream>

int main()
{
    std::cout << "Hello World!\n"; 
}

这就是我为头文件设置目录的方式：

 $(SolutionDir)libtorch\include

...和其他库的链接器目录：

$(SolutionDir)libtorch\lib

...以及我添加的所有 .lib 文件：

torch.lib;onnxifi_loader.lib;onnxifi_dummy.lib;onnx_proto.lib;onnx.lib;libprotoc.lib;libprotobuf-lite.lib;libprotobuf.lib;foxi_loader.lib;foxi_dummy.lib;cpuinfo.lib;clog.lib;caffe2_module_test_dynamic.lib;caffe2_detectron_ops.lib;caffe2.lib;c10.lib;

并不是这个库中的某些函数或类不起作用。我无法编译一个简单的“Hello World”程序。

我什至下载了一些随机的 .dll 文件来检查是否是这个特定的库问题，而其他 .dll 则没有问题。

我手动设置了项目，并按照我发送的链接中的说明进行操作（使用 CMAKE 创建项目），但仍然出现此错误。

我正在研究它几个小时，对此我感到非常恼火。我的想法已经用完了。我真的不知道在这种情况下我还能错过什么。

我在 Windows 10 上运行 VS 2017 Community，版本 15.9.14。

标签： c++pytorchlibraries

解决方案

好的，我实际上自己想出了解决方案。

由于某些我完全不明白的原因，我不得不将 .dll 文件放在我的项目文件夹中。在链接器属性中为其他库设置路径似乎不适用于这些库，至少在我的 PC 上是这样。

这非常令人困惑，因为我下载用于测试的其他随机库，我可以放在任何我想要的地方，我需要做的只是在链接器属性中设置正确的路径。

但不是这个特定的库（libtorch）。

无论如何，问题解决了，希望有一天有人发现这个有用:)

推荐阅读