600字范文,内容丰富有趣,生活中的好帮手!
600字范文 > caffe使用过程中遇到的问题和解决办法

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

时间:2023-01-15 18:25:47

相关推荐

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

cuda_success error

问题描述

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

解决方法

参考网址:/rbgirshick/rcnn/issues/28

当时在写Makefile.config时,把

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

123

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

* 最后的结果

out of memory的问题

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

使用过程中突然掉电

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

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

12参考链接://05/19/caffe-solverstate/

python接口的使用问题

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

找不到caffe.pb.h的问题

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

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

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

1234版权声明:本文为博主原创文章,未经博主允许。 /u012526003/article/details/68215496

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。