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_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代码报错解决。

你可能感兴趣的:(python,计算机视觉,目标检测,深度学习,人工智能)