ubuntu18.04下安装mmdetection 及问题解决
ubuntu18.04下安装mmdetection 及问题解决官方地址环境需求安装过程及问题官方地址mmdetection集成了多种目标检测算法,详见官方github地址,官方文档中也有安装说明,但是会遇到一些问题,博主将自己的安装过程和遇到的问题列在下面,仅供参考。环境需求RequirementsLinux (Windows is not officially supported)P...
官方地址
mmdetection集成了多种目标检测算法,详见官方github地址,官方文档中也有安装说明,但是会遇到一些问题,博主将自己的安装过程和遇到的问题列在下面,仅供参考。
环境需求
Requirements
Linux (Windows is not officially supported)
Python 3.5+ (Python 2 is not supported)
PyTorch 1.1 or higher
CUDA 9.0 or higher
NCCL 2
GCC 4.9 or higher
mmcv
We have tested the following versions of OS and softwares:
OS: Ubuntu 16.04/18.04 and CentOS 7.2
CUDA: 9.0/9.2/10.0
NCCL: 2.1.15/2.2.13/2.3.7/2.4.2
GCC: 4.9/5.3/5.4/7.3
cuda和anaconda的安装不在本文描述范围内。基础环境配置可参见Ubuntu 18.04: Install TensorFlow and Keras for Deep Learning
安装过程及问题
- 创建并激活conda环境
conda create -n mmdetection python=3.7
conda activate mmdetection
conda install cython
- 安装pytorch及torchvision
官方提供的建议是
conda install pytorch torchvision -c pytorch
但是这里有个坑,就是下载速度过慢,博主为了这个问题搞破了脑袋,试过国内镜像,科学上网,离线安装。还是国内镜像最方便,但是需要注意的是,添加国内镜像后,需要将代码改为如下所示(cudatoolkit根据cuda版本选择):
conda install pytorch torchvision cudatoolkit=9.0
不然pytorch / torchvision下载源依然为 pytorch 官网
- 安装mmcv
git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install .
- 安装mmdetection
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
python setup.py develop
在执行python setup.py develop
遇见gcc版本造成的错误。使用gcc7.4.0, gcc6.5.0均不能正常执行,后改为gcc5.5.0解决该问题,原因不详,大家可以直接使用官方测试过的gcc版本:4.9/5.3/5.4/7.3。
gcc安装及版本转化如下(以gcc6转gcc5为例):
sudo apt-get install gcc-5 g++-5
cd /usr/bin
sudo rm gcc6 g++6
sudo ln gcc-5 gcc
sudo ln g++-5 g++
可以通过以下代码查看gcc/g++版本:
gcc --version
g++ --version
参考
- https://blog.csdn.net/watermelon1123/article/details/88122020
更多推荐
所有评论(0)