编译 py-faster-rcnn
Author : Liqi
Date : 2018.08.23
E-mail : liqisa@foxmail.com
1. 本机配置
- Ubuntu16.04
- CUDA 9.0 & CUDNN7.1.3
- 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
目录是空的,则需要继续下载,执行如下指令:
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