mmdetection报错:ImportError: /home/user/repos/mmdetection/mmdet/ops/dcn/deform_conv_cuda.cpython-37m-x
环境配置:torch 1.11.0+cuda 11.3 (最新)使用mmdetection进行推断:from mmdet.apis import init_detector, inference_detector报错如下:ImportError: /home/user/repos/mmdetection/mmdet/ops/dcn/deform_conv_cuda.cpython-37m-x86_
·
环境配置:torch 1.11.0+cuda 11.3 (最新)
使用mmdetection进行推断:
from mmdet.apis import init_detector, inference_detector
报错如下:
ImportError: /home/user/repos/mmdetection/mmdet/ops/dcn/deform_conv_cuda.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN6caffe26detail37_typeMetaDataInstance_preallocated_32E
问题已解决,报错原因是pytorch版本太新导致的,虽然openmmlab支持最新版但依然会导致报错。
解决方法:
将pytorch版本退化为torch 1.6.0+cu102,查询openmmlab官方github,卸载并重装mmcv 1.3.9,重新运行mmdetection代码报错解决。
欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
已为社区贡献1条内容
所有评论(0)