Openmmlab确实是个宝藏仓库,下面简要介绍windows下安装mmcv-full的过程。
参考mmcv的官方文档(https://github.com/open-mmlab/mmcv)
重要的是问题是 需要GPU cuda, pytorch 版本一致,与mmcv对应才行。
步骤如下:
1. 安装合适版本的cuda (此处不再介绍,自行百度):本博主的cuda是10.1
2. 安装torch, torchvision版本:建议torch 1.6.0以上,本文测试了 torch1.7.0和1.8.0,都没有问题。相应的whl预编译版本下载连接,注意匹配cuda的版本,以及python版本:
https://download.pytorch.org/whl/torch_stable.html
3. 安装合适版本的mmcv-full, 如win_amd64版本,对应python 36,即可
https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.0/index.html
4. 至此已成功安装,过程中如果遇到opencv-python安装失败的,请