ModuleNotFoundError: No module named ‘mmcv._ext‘

目录

1. 更改mmcv安装方式 

1.1 严格按照官方推荐版本进行安装

1.2 下载whl文件安装

1.3 编译mmcv源码安装

 2. 测试分割网络

---

1. 更改mmcv安装方式 

这里有三种安装方式可以尝试。

1.1 严格按照官方推荐版本进行安装

打开：https://github.com/open-mmlab/mmcv

点击下图作者已经建立好的对应版本，进行安装。

我的是torch1.6. cuda10.2：

pip install mmcv-full==1.1.5 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.6.0/index.html

1.2 下载whl文件安装

到官网把whl下载下来，离线安装（建议使用）

官网1：https://download.openmmlab.com/mmcv/dist/index.htmlhttps://download.openmmlab.com/mmcv/dist/index.html

我的环境是cuda1.6.0+cuda10.2+python3.7，符合的有很多，选一个较新的版本下载。

win版本

很多人找不到win安装包，这里统一说明下，希望能够帮到大家，别把时间浪费在这里。

然后，pip install *.whl即可

1.3 编译mmcv源码安装

特别是win版本，建议源码编译，因为win版本更新到1.1.5就没有更新了。

mmcv安装记录_一千零一夜的博客-CSDN博客由于官方提供的win版本有限，需要自己源码编译mmcv，特此记录。1 准备提前下载好mmcv-master.zip或者gi clone下来：GitHub - open-mmlab/mmcv: OpenMMLab Computer Vision FoundationOpenMMLab Computer Vision Foundation. Contribute to open-mmlab/mmcv development by creating an account on GitHub.htthttps://blog.csdn.net/jizhidexiaoming/article/details/121402107或者参考官网Build MMCV from source — mmcv 1.3.17 documentationhttps://mmcv.readthedocs.io/en/latest/get_started/build.html

 2. 测试分割网络

python demo/image_demo.py demo/demo.jpg configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py \\
    checkpoints/pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth --device cuda:0 --palette cityscapes

效果：