如果官方教程不行再參考我的吧,我的環境如下:
- 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
參考:
- 官方教程:https://github.com/open-mmlab/mmdetection/blob/master/INSTALL.md
- https://blog.csdn.net/hajlyx/article/details/83542167