安装mmcv
源码地址:
https://github.com/open-mmlab/mmcv
对应表格确认自己环境的pytorch版本和cuda版本
按照指定命令安装
注意!!!
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
其中的{mmcv_version}要替换为具体的版本号
如:
pip install mmcv-full==1.3.9 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html
安装mmseg
虽然代码中用的是import mmseg
但是安装的时候mmseg要写全
命令为:
pip install mmsegmentation