Caffe入门:常见错误

1、Check failed: error == cudaSuccess (2 vs. 0) out of memory】
修改batch_size

2、Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM
在卷积层convolution_param中添加engine: CAFFE

3、“Incorrect data field size”
在生成均值文件时可能遇到该错误,因此在进行数据转换时RESIZE要设置成true.

4、libcaffe.so.1.0.0 symbol cudnnSetActivationDescriptor, version libcudnn.so.7 not defined in file libcudnn.so.7 with link time reference
重新安装cudnn:http://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html

发布了185 篇原创文章 · 获赞 873 · 访问量 127万+

猜你喜欢

转载自blog.csdn.net/duan19920101/article/details/103233164