windows10上安装mmdetection2.5.0 + pytorch1.6 + CUDA10.1 (python==3.6)

之前只能在windows上安装py37版本的, 目前已支持py36了.(2020-10-19)
能不能安装, 主要是看能装什么版本的mmcv-full.
在mmcv的安装包网址上可以看到, 目前已经有mmcv_full-1.1.5+torch1.6.0+cu101-cp36-cp36m-win_amd64.whl的安装包了.
而在mmdetection的官方安装教程中可以看到, mmdetection>=2.3.0版本是需要mmcv-full>=1.0.2的.
能够直接用pip安装mmcv-full, 这样一来, 安装就很简单了.

windows10上安装mmdetection2.5.0 + pytorch1.6 + CUDA10.1 (python==3.6)



以下是教程:

0. conda create -n mmdetection python=3.6

1. 安装pytorch
    conda install pytorch=1.6.0 torchvision cudatoolkit=10.1 -c pytorch

2. 安装mmcv-full
    pip install mmcv-full==1.1.5+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html
    或者
    pip install mmcv-full==latest+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html

3. pip install Cython

4. 下载源码
从官方git release下载: https://codeload.github.com/open-mmlab/mmdetection/zip/v2.5.0
或者git clone https://github.com/open-mmlab/mmdetection.git (截止2020-10-19, 最新版本是2.5.0)
cd mmdetection

5. 安装必须包
cd mmdetection
pip install -r requirements/build.txt

6. 编译mmdetection源码
python setup.py install develop (这个安装后,会把代码复制到python site-packages下面)
or
python setup.py build develop (推荐用这个,安装后使用的是所下载的mmdetection目录的代码, 方便修改

7. 测试
python    
import mmdet
import mmdet.core
print(mmdet.__version__)