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, 这样一来, 安装就很简单了.



以下是教程:

0. conda create -n mmdetection python=3.6

1. 安装pytorch (这里要确保cudatoolkit和已经安装的CUDA版本一致, 否则在以后使用可能会报错, 比如使用deformable conv)
    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__)





 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值