caffe 随着训练次数叠加出现out of memeory

常常会有这种情况,我们用别人的caffe 但是和cudnn 不匹配,这时,我们会忽略掉 cudnn,进行编译,开始很顺利,但是随着训练次数的叠加,会出现out of memeory 情况,原因就是我们没有用cudnn。

解决办法:

1.将./include/caffe/util/cudnn.hpp 换成最新版的caffe里的cudnn的实现,即相应的cudnn.hpp.

2. 将./include/caffe/layers里的,所有以cudnn开头的文件,例如cudnn_conv_layer.hpp。   都替换成最新版的caffe里的相应的同名文件。

3.将./src/caffe/layer里的,所有以cudnn开头的文件,例如cudnn_lrn_layer.cu,cudnn_pooling_layer.cpp,cudnn_sigmoid_layer.cu。

   都替换成最新版的caffe里的相应的同名文件。

这样,我们cudnn 就会和caffe 版本匹配。


你可能感兴趣的:(科研,deep,learning,caffe)