第一步：安裝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),一塊交流與探討,共同成長和進步!