使用目标检测开源工具MMDetection时,出现如下报错:
ModuleNotFoundError: No module named ‘mmcv._ext’
很有可能是你在开始安装mmcv-full的时候,没有指定版本,选择直接安装,如下:
pip install mmcv-full
采用这样默认安装mmcv-full的方式,如果与你环境里的cuda和torch版本不匹配,就容易出现上面报错
卸载掉原来的mmcv
pip uninstall mmcv-full
重新安装正确版本的mmcv-full
其中,{cu_version}、{torch_version}
分别对应cuda
和torch
的版本号
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
例如,我安装的是cuda10.2、pytorch1.8.0
,所以应该输入命令:
pip install mmcv-full==1.2.4 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html