官方已出最新版本可以直接在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 的安装