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

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐