找不到<torch / extension.h>

Question

我创建一个火炬扩展，然后创建this link 却抛出了

的错误

  

严重错误：火炬/扩展名。h：没有此类文件或目录。

ubuntu 18.04
code::blocks 17.04
gcc 7.4.0

#include <torch/extension.h>

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

Answer 1

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