目录
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下载下来,离线安装(建议使用)
我的环境是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 documentation
https://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
效果: