faster rcnn 中如何使用 cudnn5.1

来源:互联网 发布:迷你忍者mac中文版 编辑:程序博客网 时间:2024/04/28 02:50

环境:

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-rcnnmake cleanmake -j8make pycaffe

参考博文:

http://blog.csdn.net/kexinmcu/article/details/53178428

http://blog.csdn.net/kkk584520/article/details/51163564

0 0