Win10安装 mmdetection 2.11.0(基于mmcv1.3.1) 及Swin-Transformer

官方已出最新版本可以直接在win下安装, 参考:久等了！Windows 平台 MMCV 的预编译包终于来了！

1. 环境准备

1.1 需要先安装如下环境

• visual studio 2015或 2019 , 安装时需勾选 使用C++的桌面开发
• CUDA 10.2 + CUDNN
• anaconda 4.6.7或以上版本

通过链接下载压缩包: https://url47.ctfile.com/f/13636947-501698929-b85e1c （访问密码：7374）
解压后放到你所想要安装到的目录, 因为实际安装的 mmcv是基于本地编译的文件, 安装完成后不可再移动, 所以安装前就要放好位置

1.2 创建conda虚拟环境, 安装 python3.7, pytorch 1.8.2, torchvision 0.9.2

打开anaconda power shell, 依次使用如下命令

conda create -n mmd python=3.7
conda activate mmd
# 安装 pytorch
pip3 install torch==1.8.2+cu102 torchvision==0.9.2+cu102 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

pip install cython  matplotlib opencv-python

pytorch 安装的命令参考官网: INSTALLING PREVIOUS VERSIONS OF PYTORCH

2. 安装mmcv

2.1 设置 MSVC 编译器

• 设置环境变量。添加 C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.27.29110\bin\Hostx86\x64 到 PATH。则 cl.exe 可以在命令行中运行, 如下所示。注意如果安装的 vs2015, 中间的路径中的 版本号及数字会有些不同, 成功与否与输入命令 cl 能否正确显示决定

(base) PS C:\Users\WRH> cl
Microsoft (R) C/C++ Optimizing Compiler Version 19.27.29111 for x64
Copyright (C) Microsoft Corporation. All rights reserved.

usage: cl [ option... ] filename... [ /link linkoption... ]

为了兼容性，我们使用 x86-hosted 以及 x64-targeted 版本，即路径中的 Hostx86\x64 。

• 因为 Pytorch 将解析 cl.exe 的输出以检查其版本，只有 utf-8 将会被识别，你可能需要将系统语言更改为英语。控制面板 -> 地区-> 管理-> 非 Unicode 来进行语言转换.

2.2 编译安装

先进入下载文件的文件夹, 再执行命令

cd mmcv-1.3.1
pip install -r .\requirements.txt

TORCH_CUDA_ARCH_LIST="6.1"这一句要根据显卡的算力来, 我这里是 1060 算力是6.1, 需要进行查表, 见 NVIDA CUDA显卡算力对照表

$env:TORCH_CUDA_ARCH_LIST="6.1"
$env:MMCV_WITH_OPS = 1
$env:MAX_JOBS = 4

python setup.py build_ext
python setup.py develop

编译过程中需要等待一段时间 , 大概10分钟

3. 安装mmdetection

# 回到上一级目录, 再安装 mmdetection
cd ..
pip install -r requirements.txt
python setup.py develop

4.安装apex

cd apex-master
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" .

5.测试是否安装完成

python demo/image_demo.py demo/demo.jpg configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py mask_rcnn_swin_tiny_patch4_window7.pth

若显示图片, 则表明已经安装成功

参考: Swin-Transformer-Object-Detection环境搭建（最新mmdetection安装） WINDOWS 下 MMCV | MMCV-full 的安装