如何在py-faster-rcnn上使用最新的cuDNN v5加速庫

最近，使用新採購的gtx1080顯示卡，需要使用cudnn-v5，但是在rbgirshick大神最初的py-faster-rcnn工程上，預設使用的是cudnn-v4，如果直接使用cudnn-v5編譯faster rcnn裡的caffe會報以下錯誤：

In file included from ./include/caffe/util/cudnn.hpp:5:0,
                 from ./include/caffe/util/device_alternate.hpp:40,
                 from ./include/caffe/common.hpp:19,
                 from src/caffe/data_reader.cpp:6:
/usr/local/cuda/include/cudnn.h:799:27: note: declared here
 cudnnStatus_t CUDNNWINAPI cudnnSetPooling2dDescriptor(

原因是在cudnn v5中修改了cudnnStatus_t CUDNNWINAPI cudnnActivationBackward等函數介面，而舊版的caffe中（例如faster rcnn裡的caffe）仍然使用原來的呼叫方式。下面使用最新版本的caffe裡面的部分檔案對faster rcnn檔案進行替換，具體方法如下：

cp caffe/inlude/caffe/layers/cudnn_* caffe-fast-rcnn/include/caffe/layers/
cp caffe/src/caffe/layers/cudnn_* caffe-fast-rcnn/src/caffe/layers/
cp caffe/include/caffe/util/cudnn.hpp caffe-fast-rcnn/include/caffe/util/

按照這個修改後，就可以編譯通過。
這裡給出我修改好的py-faster-rcnn在github上的連結，不想親自修改或者怕出錯的可以直接下載。

本文永久更新連結地址：http://www.linuxidc.com/Linux/2017-10/147610.htm