本项目是作者在安装Maskrcnn/Faster-rcnn时使用高版本torch 遇到的THC.h的错误总结
问题:THC相关的包很多都在pytorch更新的过程中被抛弃了,导致很多.cu文件在安装依赖项时出现了不匹配的问题
fatal error: THC/THC.h: No such file or directory
在/maskrcnn_benchmark/csrc/cuda文件夹中的所有以.cu结尾的代码文件中删除了下述头文件:
#include <THC/THC.h>
并且把所有的:
THCudaCheck(cudaGetLastError());
替换成:
AT_CUDA_CHECK(cudaGetLastError());
“THCudaCheck“ is undefined
将所有文件中的:
THCudaCheck
替换成
C10_CUDA_CHECK
"THCCeilDiv" is undefined
将所有文件中的:
THCCeilDiv
替换成:
std::min
例如:
dim3 grid(std::min(THCCeilDiv(output_size, 512L

文章讲述了在安装Maskrcnn/Faster-rcnn时遇到THC.h相关错误,如缺失文件、未定义函数等,提供了包括替换头文件、调整内存管理以及修复torch._six模块等问题的详细解决方案。
最低0.47元/天 解锁文章
1410

被折叠的 条评论
为什么被折叠?



