笔记"/>
linux 安装mmdetection 踩坑笔记
官网链接：
说明文档：/
预训练模型：.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官网：/
conda install pytorch=1.0 torchvision -c pytorch
3、安装 mmcv
mmcv是mmdetection的一个CV库，提供了很多基本功能，后续你也可以根据自己的需要对这个库进行修改，但是每次修改好你都需要pip install . , 否则修改不起作用 。
git clone .git
cd mmcv
pip install .
4、下载 mmdetection
# 若git速度慢 ，可手动下载解压，网址：
git clone .git# 打开 mmdetection 所在文件夹
cd mmdetection
5、安装Cython，opencv及requirements.txt中的相关依赖
！ 注意【 安装顺序 】
由于限速问题，下载速度很慢，建议换成国内清华源或者中科大源。如果还是不行 ，只能离线安装，建议到 PyPI 手动下载需要的whl或tar文件，在你刚刚建立的环境下进行pip install xxx.whl进行安装。
如果不安装，在后面编译的时候会自动安装，速度巨慢，容易报错，运行一半就卡住了。如果你的网很快，就当我没说。
# 换源
sudo gedit ~/.condarc
# 添加以下源，并保存
channels:- 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里面找。问题基本都能解决
更多推荐
linux 安装mmdetection 踩坑笔记
发布评论