从C ++调用PyTorch时发生错误

Question

我尝试使用C ++中的PyTorch，但无法正常工作。命令import torch在命令提示符下效果很好。我不确定原因，但是即使在不使用CUDA的情况下安装割炬时，即使使用c ++也可以使用。

这是代码。

#include <Python.h>

int main(int argc, char** argv)
{
    Py_Initialize();
    PyRun_SimpleString("import torch");
    Py_Finalize();
    return 0;
}

请注意，它适用于例如PyRun_SimpleString("import numpy as np");。这是一条错误消息。

Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "C:\Users\GreenTea\Anaconda3\envs\pytorch\lib\site-packages\torch\__init__.py", line 81, in <module>
    from torch._C import *
ImportError: DLL load failed: The operating system cannot run %1.

我正在使用火炬== 1.2.0 + cu92火炬视觉== 0.4.0 + cu92古巴== 9.2水蟒== 5.1.0Windows == 10Visual Studio == 2019年我也尝试了其他组合。谢谢！

Answer 1

如果将pytorch与C++一起使用，则有相应的API，请参见PyTorch C++ API docs。>

[目前，它处于beta阶段，可能会发生重大变化，因此请注意，但是用法与Python API非常相似，并且经过深思熟虑，因此，如果您知道一些C++，应该不会有太大的麻烦（请参阅End to End example。）>

[如果您想使用经过训练的网络或在Python环境中感觉更好，则可以使用它来训练网络，仅通过加载用C++编译的网络就可以使用torchscript API进行推理。请参见this official tutorial，以获取易于遵循的示例操作方法。

我认为您尝试做的事情不受支持，如果您进一步采用这种方法，可能很难获得帮助。