你好,游客 登录 注册 搜索
背景:
阅读新闻

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

[日期:2017-10-14] 来源:Linux社区  作者:kexinmcu [字体: ]

最近,使用新采购的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

linux
本文评论   查看全部评论 (0)
表情: 表情 姓名: 字数

       

评论声明
  • 尊重网上道德,遵守中华人民共和国的各项有关法律法规
  • 承担一切因您的行为而直接或间接导致的民事或刑事法律责任
  • 本站管理人员有权保留或删除其管辖留言中的任意内容
  • 本站有权在网站内转载或引用您的评论
  • 参与本评论即表明您已经阅读并接受上述条款