faster rcnn在ubuntu环境下用GPU模式并使用cuDNN v5加速
本人水平低,请大家见谅! 这断时间跳进了faster rcnn这个坑里面,差点没爬起来。这篇文章主要讲在faster rcnn配置成中让cuDNN v5跑起来。主要讲gpu模式中,用cudnn v5加速,关于其余的配置问题,建议大家仔细阅读官网资料,https://github.com/rbgirshick/py-faster-rcnn。毕竟每个人的电脑配置都不同。
本人用的是CUDA8.0,cuDNN v5.1.首先要做到: 1.确认自己编译没有问题(就是不用cuDNN的时候能编译通过) 2.CUDA8.0,cuDNN v5安装好了,并且没有问题。 原文地址: https://github.com/ShuangLI59/person_search/pull/8/commits/622ff2b2f4a20bbc91f4828f3e85ffd6f22d1c75#diff-a3c8b6ea171736e9ddcf84d93b7d6364R75 废话不多说,假如要用CUDNN,需要在makefile.config中使下面选项打开: USE_CUDNN := 1 而后就需要修改下列文件: 1: caffe-fast-rcnn/include/caffe/layers/cudnn_relu_layer.hpp 在第40行+(表示增加)
2: caffe-fast-rcnn/include/caffe/layers/cudnn_sigmoid_layer.hpp
在第40行+
3: caffe-fast-rcnn/include/caffe/layers/cudnn_tanh_layer.hpp
在第40行+
4: caffe-fast-rcnn/include/caffe/util/cudnn.hpp
在第94行+,96-(’-‘表示删除)新生成的96行及 后面+
删除131,132行,并重新增内容
5: caffe-fast-rcnn/src/caffe/layers/cudnn_conv_layer.cu
删除33行,38-44行
在第77行删除并增加:
在第95行删除并增加:
6: caffe-fast-rcnn/src/caffe/layers/cudnn_relu_layer.cpp 在16行增加
7: caffe-fast-rcnn/src/caffe/layers/cudnn_relu_layer.cu
在第18行,20行,25行删除增加:
在52,54,60行删除增加:
8: caffe-fast-rcnn/src/caffe/layers/cudnn_sigmoid_layer.cpp
15,16行增加:
9: caffe-fast-rcnn/src/caffe/layers/cudnn_sigmoid_layer.cu 在13,15,20行增加删除:
在42,44,50增加删除:
10: caffe-fast-rcnn/src/caffe/layers/cudnn_tanh_layer.cpp
在16行增加:
11: caffe-fast-rcnn/src/caffe/layers/cudnn_tanh_layer.cu
在13,15,20行增加删除:
在43,45,51行增加删除:
而后重新编译makefile.config文件(记住先make clean后make)
后面我碰到的一个问题是 ‘numpy.float64’ object cannot be interpreted as an index 主要起因是版本不支持问题,处理方法:sudo pip install -U numpy==1.11.0
希望大家能把GPU模式跑起来,毕竟cpu和gpu不是一个数量级的.cudnn还能够减少消耗的显存。 有什么安装,编译问题大家可以一起讨论!

1. 本站所有资源来源于用户上传和网络,如有侵权请邮件联系站长!
2. 分享目的仅供大家学习和交流,您必须在下载后24小时内删除!
3. 不得使用于非法商业用途,不得违反国家法律。否则后果自负!
4. 本站提供的源码、模板、插件等等其他资源,都不包含技术服务请大家谅解!
5. 如有链接无法下载、失效或广告,请联系管理员处理!
6. 本站资源售价只是摆设,本站源码仅提供给会员学习使用!
7. 如遇到加密压缩包,请使用360解压,如遇到无法解压的请联系管理员
开心源码网 » faster rcnn在ubuntu环境下用GPU模式并使用cuDNN v5加速