mmdetection安装教程

如果官方教程不行再参考我的吧,我的环境如下:

  • ubuntu
  • cuda10
  • cudnn7.5

步骤:

1.使用conda创建一个虚拟环境

conda create -n mmdetection python=3.7
conda activate mmdetection

2.安装Pytorch(2019.7.24安装的是Pytorch1.1)

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

3.安装Cython

conda install cython

4.安装mmcv

mmcv也是mmdetection团队开发的一个CV库,提供了很多基本功能,所以需要先安装这个。后续你也可以根据自己的需要对这个库进行修改,但是每次修改好你都需要使用如下命令重新安装mmcv,否则你的修改是起不了作用的。

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install .

5.安装mmdetection

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
./compile.sh
python setup.py install

参考:


微信公众号:AutoML机器学习
MARSGGBO原创
如有意合作或学术讨论欢迎私戳联系~
邮箱:marsggbo@foxmail.com





2019-7-24



原文地址:https://www.cnblogs.com/marsggbo/p/11240534.html