环境:
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