您的位置:首页 > 其它

ubuntu16.04 cuda8.0 cudnn6.0 faster-rcnn配置

2017-09-06 21:19 806 查看
参考:
http://blog.csdn.net/nicky_lyu/article/details/53181434 http://blog.csdn.net/u012841667/article/details/53436615 http://www.cnblogs.com/zjutzz/p/6034408.html
step 1. 安装常用的dependencies

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler libhdf5-dev

sudo apt-get install –no-install-recommends libboost-all-dev

sudo apt-get install libatlas-base-dev

sudo apt-get install python-dev

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

sudo pip install cython 
sudo pip install easydict 

先占坑 ,等我装完了写



解决办法:

1).将/py-faster-rcnn/caffe-fast-rcnn/include/caffe/util/cudnn.hpp 换成最新版的caffe里的cudnn的实现,即相应的cudnn.hpp.

2).将/py-faster-rcnn/caffe-fast-rcnn/src/caffe/layer里的,所有以cudnn开头的文件,例如cudnn_lrn_layer.cu,cudnn_pooling_layer.cpp,cudnn_sigmoid_layer.cu。

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

src/caffe/layers/cudnn_relu_layer.cpp: In member function ‘virtual void caffe::CuDNNReLULayer<Dtype>::LayerSetUp(const std::vector<caffe::Blob<Dtype>*>&, const std::vector<caffe::Blob<Dtype>*>&)’:

src/caffe/layers/cudnn_relu_layer.cpp:16:45: error: ‘activ_desc_’ was not declared in this scope

cudnn::createActivationDescriptor<Dtype>(&activ_desc_, CUDNN_ACTIVATION_RELU);

^

make: *** [.build_release/src/caffe/layers/cudnn_relu_layer.o] 错误 1

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

运行faster-rcnn里的demo

cd py-faster-rcnn/tools  

./tools/demo.py  



我出现了一个问题,说需要安装 python-tk

用以下命令安装

 sudo apt-get install python-tk

----------------------------更新----------------------------------------------

在用faster rcnn训练时 遇到错误

pb2.text_format.Merge(f.read(), self.solver_param) AttributeError: 'module' object has no attribute 'text_format'

经过google之后发现是protobuf的本版发生了变换,之前在配置caffe的时候手动安装了protbuf,版本是2.5.0,后来安装了tensorflow 我回忆了一下,protobuf的版本貌似是发生了变换。

所以解决办法:sudo pip install protobuf==2.5.0

刚才又发现一个新的解决方法:

在文件./lib/fast_rcnn/train.py增加一行import google.protobuf.text_format 即可解决问题

faster rcnn 训练  遇到 numpy.core.multiarray 错误

解决办法:

更新到最新的numpy版本 pip install -U numpy

又遇到新的问题:

numpy.float64' object cannot be interpreted as an index

解决办法
http://m.blog.csdn.net/hongbin_xu/article/details/77278329
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐