matlab中运行caffe接口net.forward_prefilled()导致matlab崩溃的问题

我先在工作站上用caffe训练好了一个用于图像超分辨的63层conv层的模型,参数比较多,然后把训练好的模型放在自己笔记本上,用matcaffe接口跑网络的前向传播做测试,测试中关于前向传播的代码如下(用gpu模式):

im_input=permute(im_l,[2,1,3]);
net.blobs('data').reshape([size(im_input),1,1]);
net.reshape();
net.blobs('data').set_data(im_input); 
net.forward_prefilled();
im_result=net.blobs('upsample').get_data();

结果发现输入较大的图像时,matlab直接崩溃!而输入较小的图像就能正常运行,而且用cpu模式可以正常运行所有图像,我笔记本内存16G,渣渣显存只有2G,所以想到是不是爆显存了。然后在测试时用gpu-z查看显存,果然!显存瞬间拉满,然后matlab崩溃!所以以后再遇到matlab中运行matcaffe接口导致matlab崩溃的问题,先想想是不是爆显存了。(辣鸡caffe都爆显存了也不给个“out of memory”的提示)

猜你喜欢

转载自blog.csdn.net/Kyrielong/article/details/82219644
今日推荐