红联Linux门户
Linux帮助

faster rcnn中如何使用cudnn5.1,无法编译通过cudnn:=1

发布时间:2017-03-19 10:30:21来源:linux网站作者:不吹牛不睡觉
环境:
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://www.linuxdiyf.com/linux/29297.html