自定义C++扩展名:找不到torc/torc.h

Question

我将按照本教程为Pytorch创建一个C++扩展。我的C++代码出现以下错误：

test.cpp:3:10: fatal error: torch/torch.h: No such file or directory
 #include <torch/torch.h>

如何获取torch.h头文件？有pytorch-dev版本吗？

Answer 1

您很可能需要安装c++版本的torch。

如果你前往www.pytorch.org，他们会引导你选择适合你的下载。只需确保选择c++版本即可。

希望这能奏效！