THC/THC.h: No such file or directory (pytorch 1.13.1)
·
我的pytorch版本为1.13.1，CUDA版本为v11.7，系统win11，在安装编译一个第三方库时报错：THC/THC.h: No such file or directory。
在网上找了一下解决方法，说是在pytorch 1.11版本开始，删去了THC相关文件，可以用ATen相关替代。首先找到报错的代码，注释#include <THC/THC.h>，再运行时，用到THC.h的代码将会报错，例如我这里的错误信息是：
error C2039: "getCurrentCUDAStream": 不是 "at::cuda" 的成员
D:\miniforge3\envs\pytorch1.13_cuda\lib\site-packages\torch\include\ATen/detail/CUDAHooksInterface.h(14): note: 参见“at::cuda”的声明
error C3861: “getCurrentCUDAStream”: 找不到标识符在pytorch的Github网页中可以找到每个版本的发布日志：Releases · pytorch/pytorch · GitHub
在发布日志中找到自己关注的版本号，根据自己的报错查找所需函数的替代方法。
对于我这里的错误信息，在代码中将 #include <THC/THC.h> 改为 #include <ATen/cuda/CUDAContext.h>，即可正常使用at::cuda::getCurrentCUDAStream()了。
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
3
0- 0
已为社区贡献2条内容
所有评论(0)