error: identifier “AT_CHECK“ is undefined

原创已于 2022-05-03 09:34:52 修改 · 1.2k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #python

于 2022-05-01 22:09:10 首次发布

深度学习专栏收录该内容

3 篇文章

订阅专栏

这篇博客分享了在使用Detectron2进行mask-rcnn项目时遇到的编译错误，通过将代码中的AT_CHECK替换为TORCH_CHECK解决了问题。博主详细解释了错误原因并提供了解决方案，适用于对Detectron2库有需求的开发者。

这两天运行一个mask-rcnn系列网络的代码，在执行命令 python3 setup.py build develop 时收到一堆这样错误如下：

.../detectron2/layers/csrc/deformable/deform_conv.h(136): error: identifier "AT_CHECK" is undefined

.../detectron2/layers/csrc/deformable/deform_conv.h(184): error: identifier "AT_CHECK" is undefined

.../detectron2/layers/csrc/deformable/deform_conv.h(234): error: identifier "AT_CHECK" is undefined

.../detectron2/layers/csrc/deformable/deform_conv.h(284): error: identifier "AT_CHECK" is undefined

.../detectron2/layers/csrc/deformable/deform_conv.h(341): error: identifier "AT_CHECK" is undefined

.../detectron2/layers/csrc/deformable/deform_conv_cuda.cu(155): error: identifier "AT_CHECK " is undefined

.../detectron2/layers/csrc/deformable/deform_conv_cuda.cu(338): error: identifier "AT_CHECK " is undefined

.../detectron2/layers/csrc/deformable/deform_conv_cuda.cu(503): error: identifier "AT_CHECK " is undefined

.../detectron2/layers/csrc/deformable/deform_conv_cuda.cu(696): error: identifier "AT_CHECK " is undefined

.../detectron2/layers/csrc/deformable/deform_conv_cuda.cu(823): error: identifier "AT_CHECK " is undefined

.../detectron2/layers/csrc/deformable/deform_conv_cuda.cu(953): error: identifier "AT_CHECK " is undefined

11 errors detected in the compilation of ".../detectron2/layers/csrc/deformable/deform_conv _cuda.cu".

百度一大堆之后找到了解决方法：把涉及错误的文件里面所有的AT_CHECK替换为TORCH_CHECK，大功告成！

参考博客：https://blog.youkuaiyun.com/weixin_44444492/article/details/118887280