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的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
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