faster-rcnn环境搭建遇到的问题

来源:互联网 发布:苹果6打不开数据流量 编辑:程序博客网 时间:2024/05/16 00:40

make的时候遇到的cudnn的问题,用这篇文章的方法解决了http://blog.csdn.net/u011070171/article/details/52292680

这是因为当前版本的caffe的cudnn实现与系统所安装的cudnn的版本不一致引起的。

解决办法:

1.将.CAFFE_ROOT/include/caffe/util/cudnn.hpp 换成最新版的caffe里的cudnn的实现,即相应的cudnn.hpp.

2. 将.CAFFE_ROOT/include/caffe/layers里的,所有以cudnn开头的文件,例如cudnn_conv_layer.hpp。   都替换成最新版的caffe里的相应的同名文件。

3.将.CAFFE_ROOT/src/caffe/layer里的,所有以cudnn开头的文件,例如cudnn_lrn_layer.cu,cudnn_pooling_layer.cpp,cudnn_sigmoid_layer.cu。

   都替换成最新版的caffe里的相应的同名文件。


rbgirshick的py-faster-rcnn实现,因为其cudnn实现为旧版本的实现,所有出现了以上问题.