faster rcnn 中如何使用 cudnn5.1

本文解决FastRCNN中Caffe无法通过CUDNN编译的问题,通过从BLVC Caffe复制CUDNN相关文件到FastRCNN目录,并进行相应清理及重新编译,使FastRCNN支持CUDNN5.1。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

环境:

ubuntu16.04.2

gtx1050 2g

cuda8

cudnn5.1

问题描述:

使用faster rcnn中的caffe时,修改Makefile.config,无法编译通过 cudnn:=1,而blvc中的caffe是可以再cudnn:=1的情况下编译成功的。原因是随之caffe的更新,支持了cudnn5.1,而faster rcnn中的caffe一直没有更新

解决办法:

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/

以上路径根据实际情况而定。

cd caffe-fast-rcnn
make clean
make -j8
make pycaffe

参考博文:

http://blog.youkuaiyun.com/kexinmcu/article/details/53178428

http://blog.youkuaiyun.com/kkk584520/article/details/51163564

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值