caffe使用过程中遇到的问题和解决办法

cuda_success error

问题描述

• 在参考http://www.cnblogs.com/denny402/p/5083300.html自己尝试预处理和训练数据的时候，出现了error == cudaSuccess (11 vs. 0) invalid argument的问题

解决方法

• 参考网址：https://github.com/rbgirshick/rcnn/issues/28

• 当时在写Makefile.config时，把

  CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
  -gencode arch=compute_20,code=sm_21 \
  

给注释了，在上面的网址中将这两行取消注释，重新编译一下即可正常训练和测试了
* 最后的结果

out of memory的问题

• 可以通过设置test的batchsize和test_iter。将batchsize调节得小一点，只要保证两个相乘之和等于测试的个数即可

使用过程中突然掉电

• 可以从caffemodel中恢复当前状态并继续训练，参考代码如下：

  ./build/tools/caffe train --solver=models/bvlc_reference_caffenet/solver.prototxt --snapshot=models/bvlc_reference_caffenet/caffenet_train_iter_10000.solverstate
  

• 参考链接：http://nfeng.cc/2016/05/19/caffe-solverstate/

python接口的使用问题

• 问题：'module' object has no attribute 'LabelMap'
• 解决办法：需要将python的目录添加到~/.bashrc中去。
• 参考链接：https://github.com/weiliu89/caffe/issues/4

找不到caffe.pb.h的问题

• 我的问题是caffe/util/cudnn.hpp:8:34: fatal error: caffe/proto/caffe.pb.h: 没有那个文件或目录

• 在网上找了很多这个问题，但是相同的问题的原因不一样，我的caffe在make与cmake的过程中都没有问题，后来是在https://github.com/muupan/dqn-in-the-caffe/issues/3上找到了解决办法，缺失的文件是放在caffe安装根目录的include文件夹中，具体的解决办法如下

  protoc src/caffe/proto/caffe.proto --cpp_out=.
  mkdir include/caffe/proto
  mv src/caffe/proto/caffe.pb.h include/caffe/proto