安装
引言
在最近的比赛中又一次使用了MMdetection库,上次使用是在天池大赛中跑baseline,这次在Ubuntu服务器下重新安装MMdetection还是处处碰壁,总结以下问题,很多问题在github issue中重复。
创建一个新环境
conda create -n name python==3.6
推荐使用3.6版本
cuda 版本
# 查看服务器上的cuda版本,强烈推荐安装10.x版本!
cat /usr/local/cuda/version.txt
安装pytorch
pip install torch==1.1.0 torchvision==0.3.0 cudatoolkit==10.0.130
为了避免错误,这是最稳定的版本配置
纠正numpy版本
pip install numpy==1.16.0
numpy v1.18会引起int 和 float64计算的不兼容,在mmdetection中表现了 test计算AP失败
MMdet支持的pytroch版本
mmdetV1.0:torch1.1,1.2,1.3,1.4
mmdetV2.0:torch1.5
要注意pytorch的版本,最近出了putorch1.5,如果不指定版本去安装都会安装成pytorch1.5,但是1.5版本是在MMdetectionV2.0中被兼容的,如果需要使用V2.0则使用以下语句进行安装
git clone https://github.com/open-mmlab/mmdetection -b v2.0
cd mmdetection
cython,wheel
pip install cython wheel
mmcv
pip install mmcv
或者
git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install -e .
如果安装mmcv出现了问题,可以尝试使用
pip install pytest-runner
这是一个必要的库
pycocoapi
pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
或者
pip install pycocotools
mmdetection v1
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -e -v .
安装失败后
python setup.py clean
手动删除build文件,在处理完问题后继续以下
pip install -v -e .
或者
python setup.py develop