个人微信公众号:AI研习图书馆,欢迎关注~
深度学习知识及资源分享,学习交流,共同进步~
编译 py-faster-rcnn
1. 本机配置
- Ubuntu16.04
- CUDA 8.0 & CUDNN5.1
- python2.7
查看CUDA和CUDNN版本的方法:
终端中输入
cuda 版本
cat /usr/local/cuda/version.txt
cudnn 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
2. 下载py-faster-rcnn源码
使用git从github上克隆源码仓库
git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git
–recursive :是为了递归下载caffe-master
因为 py-faster-rcnn/caffe-fast-rcnn来自另一个仓库
注: 如果执行完git clone指令后发现 py-faster-rcnn/caffe-fast-rcnn目录是空的,则需要继续下载,执行如下指令:
```python
cd caffe-fast-rcnn
git submodule update --init --recursive
3. 安装依赖项
进入py-faster-rcnn/caffe-fast-rcnn/python目录,caffe-fast-rcnn/python/requirement.txt文件列出了pycaffe的所有依赖库,要把requirements下的依赖都装一遍,执行如下指令:
$for req in $(cat requirements.txt); do pip install $req; done
4. 编译Cython
进入py-faster-rcnn/lib目录,执行如下指令:
make
5. 编译caffe
更新caffe版本
需要先更新用到的caffe,原版py-faster-rcnn依赖的caffe比较老,不支持cudnnv5,那就更新下对应的源码,进入 py-faster-rcnn/caffe-fast-rcnn 依次执行如下指令:
cd caffe-fast-rcnn
git remote add caffe https://github.com/BVLC/caffe.git
git fetch caffe
git merge -X theirs caffe/