环境配置: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代码报错解决。