进来实验室卡空着,想着以前只是阅读过mmdetection的代码,想着跑着试试
然后我感觉我安装过程中遇到了无数坑。
首先,实验室的卡是TITAN RTX的卡,但是由于某些原因,装了cuda9,所以这边考虑安装的是
- pytorch 1.1 cuda92
- torchvision 0.3
建议大家还是安装anaconda,这边我们新建一个虚拟环境
conda create -n mmdet python=3.6
source activate mmdet
pip install cython
建议这边使用清华的源安装anaconda,这样会快一点
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
开始安装pytorch
conda install pytorch=1.1 cuda92
conda install torchvision
然后呢,我这边没有检查就一直安装安装到最后,发现torchvison加载不进来,查了最后,是因为pillow升级了,7.0的移除了一个函数,说pytorch1.4解决了这个问题,然后接着重新安装
conda install pytorch=1.4 cuda92
conda install torchvision
然后安装mmcv
git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install -e .
但是在check opencv-python的时候,死活check不过去,但是我这边是安装了opencv的,想了想,是以为我用的menpo装的,
conda install -c menpo opencv
其实这边不管这个也行,但是一时钻了牛角尖,算了,就改用pip装,pip说一直找不到,然后查了好久,说改个源就行
cd
然后继续安装mmcv,然后
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
这边所有的按照官网上来就行
pip install -r requirements/build.txt
pip install "git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI"
pip install -v -e . # or "python setup.py develop"
在运行到最后一步的时候,神奇的是,编译失败了,当时差点崩溃
于是想到可能是cuda不兼容,然后又回到最终的起来,安装了低版本的pillow,继续安装pytorch1.1,
然后跑了一下,这位老哥的教程
努力简约起来:安装配置mmdetectionzhuanlan.zhihu.com![95274504264477cdf0befc0dafe86fa1.png](https://i-blog.csdnimg.cn/blog_migrate/399dff38e1f824211ebaf933b2e09082.jpeg)
最终终于搞定了
总结一下就是先把conda和pip添加清华的镜像
conda create -n mmdet python=3.6
source activate mmdet
pip install cython
pip install pillow=6.1
conda install pytorch=1.4 cuda92
conda install torchvision
git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install -e .
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install "git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI"
pip install -v -e . # or "python setup.py develop"
然后就大功告成了