caffe 常见错误
一颗行走的大白菜
本人专注于计算机视觉和深度学习的应用
展开
-
caffe error :Check failed: error == cudaSuccess (2 vs. 0) out of memory
用caffe 训练网络时,遇到了Check failed: error == cudaSuccess (2 vs. 0) out of memory意思是gpu内寸不够用,输入:nvidia-smi可以看到最后两个线程占用gpu内存很大。需要使用命令:sudo kill -9 [线程pid]把不需要的线程杀死,然后就能愉快的训练了...原创 2018-05-29 12:42:39 · 1486 阅读 · 0 评论 -
caffe 与cudnn版本不匹配 ./include/caffe/util/cudnn.hpp:127:41: error: too few arguments to function
这是因为当前版本的caffe的cudnn实现与系统所安装的cudnn的版本不一致引起的。解决办法:1.将./include/caffe/util/cudnn.hpp 换成最新版的caffe里的cudnn的实现,即相应的cudnn.hpp.将./include/caffe/layers里的,所有以cudnn开头的文件,例如cudnn_conv_layer.hpp。 都替换成最新版的caf...转载 2018-09-29 10:13:04 · 813 阅读 · 0 评论