caffe学习笔记:windows环境caffe使用错误锦集

写在前面:小白入门深度学习框架caffe,其间遭遇各种Check failed,特做整理汇总。问题大多都可以在网上找到原因,没有的则通过阅读源码解决(建议遇到问题找到源码,大多有提示)。

1. Check failed: net_->num_intputs() == 1 Network should have exactly one intput

原因:使用train_val.prototxt而未使用deploy.deploy.prototxt做预测,注意测试用模型文件与训练用模型文件不能用同一个,需要细微改动。

2. Check failed: status == CUDNN_STATUS_SUCCESS(6 vs. 0) CUDNN_STATUS_MISMATCH

原因:显卡的compute capability不够,只有2.1,而cuDnn需要3.0以上,在这里可以查看你的gpu是否支持cudnn, https://developer.nvidia.com/cuda-gpus。

3. Check failed: ReadProtoFromTextFile(param_file, param) Failed to parse NetParameter

原因:网络模型文件有识别不了的字符,应注意在layer与layers写网络类型type时略有不同,具体请查看caffe学习笔记:caffe网络模型文件(.prototxt)layer与layers不同。

4. Check failed:*ptr host allocation of size failed

原因:机器配置较低,如内存不够用等,可将batch_size调小,使内存使用量减小试试。

你可能感兴趣的:(caffe学习)