windows10训练pixelnet 时matlab中运行caffe接口net.forward_prefilled()导致matlab崩溃的问题

情况说明:

电脑配有英伟达GTX1080Ti的显卡,显存11G。
caffe编译通过。
matlab训练pixelnet时,运行startup.m---->trainSeg.m---->的trainNet函数到trainModel函数到solver.net.forward_prefilled();单步运行到Net.m中的forward_prefilled(self)中出现崩溃现象。
并弹出提示框(有的matlab可能不会弹框直接闪退(因为这个MATLAB System Error 有一个never send选项,如果勾选过了就不在提示了)):
windows10训练pixelnet 时matlab中运行caffe接口net.forward_prefilled()导致matlab崩溃的问题_第1张图片
原因分析:
在caffe框架搭好之后进行了minist训练,训练成功,则证明matlab接口没有问题。
在整个过程中我查看显存是没有陡然上升的,因此感觉也不是显存的问题。
当训练模式选为cpu时,即在训练时caffe.set_mode_cpu,网络可以正常跑通。
因此感觉是显卡有问题,但是minist训练时的模式就是GPU训练,没有问题。

总结:
在另一台同样配置的服务器上训练、pixelnet代码相同、caffe相同、matlab。cuda、cudnn版本相同可以通过训练。
但是在这台服务器上就是不行。最终个人认为是caffe生成的matlab接口有问题。还有就是版本不兼容的问题。

作者:GL3_24
来源:CSDN
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

你可能感兴趣的:(caffe)