mmsegmentation1.2.2+mmcv的安装(Win10)

目前网上大多数的安装教程都是针对mmsegmentation 0.x版本的，但是官方的版本已经更新到了1.x，而且1.x与0.x之间是不兼容的，所以这里记录一下1.2.2版本的安装过程，包括对应的mmcv的安装。

首先，安装过程主要是按照mmsegmentation官方GitHub仓库的说明进行的，但是官方的教程中没有C++编译器、CUDA、cudnn的安装说明，而这些对于最终的成功安装是非常重要的。

首先，需要下载VS2019，记得安装的时候一定要选择C++开发环境，安装后之后在环境变量里编辑系统变量的Path，添加上c++编译器的路径，D:\Program Files\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30133\bin\Hostx86\x64

添加环境变量之后cmd命令行窗口输入cl，如果显示cl的版本信息，则说明c++编译器环境配置成功。

接下来要确定安装的CUDA版本和Pytorch版本，这里我使用了CUDA11.7，和Pytorch1.13。CUDA下载连接，选择要下载的版本，然后在下载页面选择系统版本，exe(local)表示下载本地安装包。

安装CUDA的时候一定要注意，如果已经安装了显卡驱动，那么一定要选择自定义安装，然后在安装选项页面那勾选CUDA，其他几项不要勾选，其他的诸如安装路径什么的，建议不要更改，如果更改的话要记着装在哪里了。CUDA安装完之后，需要下载cudnn，下载链接，我安装的是CUDA11.7，所以下载对应版本的cudnn，cudnn下载后解压，解压后目录下的内容全部复制，然后进入CUDA的根目录，我的是默认的路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7，将复制的文件粘贴到这里。通常情况下，在安装CUDA的过程中会自动设置环境变量，这时候可以打开cmd，输入nvcc -V,如果显示CUDA的版本信息则说明CUDA安装成功。

安装好CUDA后创建conda虚拟环境(需要安装anaconda)，打开anaconda的powershell，如下图，注意不要用cmd，容易安装失败。在powershell中创建虚拟环境：

conda create --name openmmlab python=3.8 -y
conda activate openmmlab

安装pytorch1.13(CUDA 11.7):

# CUDA 11.7
conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.7 -c pytorch -c nvidia

使用MIM 安装 MMCV（MIM是open-mmlab的官方包管理工具）

pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"

安装MMCV这一步非常重要，直接决定了最后是否成功，因为1.x的mmsegmentation支持的是2.0以上版本的mmcv，所以这里要指定mmcv>=2.0.0, 另外2.0.0以后的mmcv对应了原先的mmcv-full，所以这里mmcv后面不用加-full了。

安装好mmcv之后就可以开始安装mmsegmentation了，这里推荐从源码安装

git clone -b main https://github.com/open-mmlab/mmsegmentation.git
cd mmsegmentation
pip install -v -e .
# '-v' 表示详细模式，更多的输出
# '-e' 表示以可编辑模式安装工程，
# 因此对代码所做的任何修改都生效，无需重新安装

如果没有报错，那么到这里就安装完成了。接下来可以验证下 

下载配置文件和权重(仍然在mmsegmentation目录下)

mim download mmsegmentation --config pspnet_r50-d8_4xb2-40k_cityscapes-512x1024 --dest .

python demo/image_demo.py demo/demo.png configs/pspnet/pspnet_r50-d8_4xb2-40k_cityscapes-512x1024.py pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth --device cuda:0 --out-file result.jpg

如果提示缺少ftfy的话可以使用pip install ftfy进行安装，运行成功后会在当前目录下生成一个result.jpg文件,说明mmsegmentation安装成功。