找不到 <torch/extension.h>

Can not find <torch/extension.h>

本文关键字:gt extension torch lt 找不到      更新时间:2023-10-16

>我创建了一个pytorch扩展,后跟这个链接但它抛出了错误

致命错误:torch/extension.h:没有这样的文件或目录。

ubuntu 18.04
code::blocks 17.04
gcc 7.4.0
#include <torch/extension.h>

我希望代码能够运行。是否有任何文档可以解决此问题?

是为 C++/Cuda 扩展创建 Python 绑定的规范头文件。因此,如果 Pytorch 版本不是最新版本,请尝试升级它。