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”的提示,最后是自己查log才发现有“out of memory”,没人带真的很容易踩坑o(╥﹏╥)o)

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值