第一步:安装python包依赖
sudo pip install Cython
sudo pip install easydict
第二步:下载源代码
git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git
在py-faster-rcnn/lib目录下修改和编译Cython
将setup.py文件第135行
'nvcc': ['-arch=sm_35',
对照NVIDIA网站 改为符合自己显卡的数字,比如GTX 1080对应61,即
'nvcc': ['-arch=sm_61',
然后编译:
make -j8
第三步:在caffe-faster-rcnn目录下修改Makefile.config文件
cp Makefile.config.example Makefile.config
注意修改
USE_CUDNN := 1
WITH_PYTHON_LAYER := 1
USE_PKG_CONFIG := 1
第四步:替换为新版CUDNN文件
由于caffe-fast-rcnn在编写时使用的是低版本的cudnn,我们需要先下载或者git clone最新版的caffe,然后再进行文件替换。
主要替换文件如下:
py-faster-rcnn/caffe-fast-rcnn/include/caffe/util/cudnn.hpp
py-faster-rcnn/caffe-fast-rcnn/src/caffe/util/cudnn.cpp
py-faster-rcnn/caffe-fast-rcnn/src/caffe/layers/ 下16个cudnn_开头的文件
py-faster-rcnn/caffe-fast-rcnn/include/caffe/layers/下8个cudnn_开头的文件
第五步:编译
make -j8 && make pycaffe
福利区:独学而无友,则孤陋而寡闻.加入机器学习与深度学习讨论QQ群(581789266),一块交流与探讨,共同成长和进步!