训练fcos代码时出现问题
ImportError: cannot import name ‘_C’
原因是没有对代码进行编译
运行python setup.py develop --no-deps进行代码编译
编译过程中出现报错:
fcos_core/csrc/cuda/ROIAlign_cuda.cu:5:10: fatal error: THC/THC.h: No such file or directory
#include <THC/THC.h>
^~~~~~~~~~~
compilation terminated
问题原因:高版本的torch中取消了这个接口
解决办法:将torch版本降到1.9.0及以下的版本
之后又遇到问题
error: identifier “AT_CHECK“ is undefined
解决办法:将报错信息中所有提到的AT_CHECK替换为TORCH_CHECK
之后在运行
python setup.py develop --no-deps
编译成功