linux 安装mmdetection 踩坑笔记

官网链接：https://github.com/open-mmlab/mmdetection
说明文档：https://mmdetection.readthedocs.io/en/latest/
预训练模型：https://github.com/open-mmlab/mmdetection/blob/master/docs/MODEL_ZOO.md

配置要求

• Linux 16
• Python 3.5+
• PyTorch 1.1 or higher
• CUDA 9.0 or higher
• NCCL 2
• GCC 4.9 or higher
• mmcv

查看方式

	命令
查看 cuda 版本	nvcc -V 或者 cat /usr/local/cuda/version.txt
查看 python 版本	python -V 或者 python
查看 gcc 版本	gcc -version
查看系统版本号	lsb_release -a
查看主板信息	lspci
查看CPU信息	cat /proc/cpuinfo
查看硬盘型号	cat /proc/ide #(IDE硬盘) 或者 cat /proc/scsi #(SCSI硬盘)
查看内存信息	cat /proc/meminfo
查看显卡信息	nvidia-smi
查看 物理cpu个数	grep 'physical id' /proc/cpuinfo | sort -u
查看 核心数量	grep 'core id' /proc/cpuinfo | sort -u | wc -l
查看 线程数	grep 'processor' /proc/cpuinfo | sort -u | wc -l

安装步骤

建议新建一个环境，否则容易出错。

1、新建环境

conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab # 或 source activate open-mmlab 

2、安装 pytorch

# pytorch官网：https://pytorch.org/get-started/locally/
conda install pytorch=1.0 torchvision -c pytorch

3、安装 mmcv

mmcv是mmdetection的一个CV库，提供了很多基本功能，后续你也可以根据自己的需要对这个库进行修改，但是每次修改好你都需要pip install . , 否则修改不起作用 。

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

4、下载 mmdetection

# 若git速度慢 ，可手动下载解压，网址：https://github.com/open-mmlab/mmdetection
git clone https://github.com/open-mmlab/mmdetection.git

# 打开 mmdetection 所在文件夹
cd mmdetection

5、安装Cython，opencv及requirements.txt中的相关依赖

！ 注意【 安装顺序 】
由于限速问题，下载速度很慢，建议换成国内清华源或者中科大源。如果还是不行 ，只能离线安装，建议到 PyPI 手动下载需要的whl或tar文件，在你刚刚建立的环境下进行pip install xxx.whl进行安装。
如果不安装，在后面编译的时候会自动安装，速度巨慢，容易报错，运行一半就卡住了。如果你的网很快，就当我没说。

# 换源
sudo gedit ~/.condarc
# 添加以下源，并保存
channels:
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
  - https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - defaults
 
# 显示源
show_channel_urls: true

6、编译

两种版本的编译方法不同，注意区别。现在大多都是1.1版本。

# PyTorch version 1.1 mmdetection 
python setup.py develop  # or "pip install -v -e ."

# PyTorch-0.4.1 version mmdetection 
./compile.sh
python setup.py install

7、遇到问题，先到github的 issues里面找。问题基本都能解决