将THC/THC.h模块替换为ATen/ATen.h模块
将THC/THC.h模块替换为ATen/ATen.h模块
提问于 2022-07-15 03:23:52
我对替换方法有疑问。最近，我正在安装用cpp和cuda编译的不同的损失函数。然而，我面对的是一个致命的错误
'THC/THC.h': No such file or directory我发现TH(C)方法目前在pytorch的最新版本中被废弃，取而代之的是ATen API (https://discuss.pytorch.org/t/question-about-thc-thc-h/147145/8)。
当然，降低我的火炬版本会解决这个问题。但是，由于我的GPU兼容性问题，我别无选择，只能自己修改脚本。因此，我的问题可归纳为以下几点。
--首先，如何使用ATen API替换依赖TH(C)方法的代码？下面是我必须修改的代码，替换这三行对于我的情况来说已经足够了。
#include <THC/THC.h>
extern THCState *state;
cudaStream_t stream = THCState_getCurrentStream(state);第二次，对cpp文件的单次修改是否足以解决我现在面临的问题？(这只是个小问题，回答第一个问题就够了)。
作为参考，我附加了我要构建的文件(https://github.com/sshaoshuai/Pointnet2.PyTorch)的github链接。
回答 1
Stack Overflow用户
回答已采纳
发布于 2022-07-15 07:43:12
在挣扎了一段时间之后，我找到了自己的答案。在THCState_getCurrentStream的情况下，它可以直接被at::cuda::getCurrentCUDAStream()取代。因此，修改后的代码块如下所示。
//Comment Out
//#include <THE/THC.h>
//extern THCState *state;
//cudaStream_t stream = THCState_getCurrentStream(state);
//Replace with
#include <ATen/cuda/CUDAContext.h>
#include <ATen/cuda/CUDAEvent.h>
cudaStream_t stream = at::cuda::getCurrentCUDAStream();在替换了整个源代码之后，我成功地构建了这个模块。
希望这能有所帮助。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/72988735
复制相关文章
相似问题
腾讯云开发者
