error: identifier “AT_CHECK“ is undefined

wangmeili@

已于 2022-05-03 09:34:52 修改

阅读量1k

点赞数 6

分类专栏：深度学习文章标签：深度学习 python

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

本文链接：https://blog.csdn.net/Retarded78_/article/details/124533486

版权

深度学习专栏收录该内容

3 篇文章 0 订阅

订阅专栏

这两天运行一个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.csdn.net/weixin_44444492/article/details/118887280