这是tensorflow 一个常常性错误,错误的缘由在于:显卡内存不够。网络
解决方法就是下降显卡的使用内存,途径有如下几种措施:图片
1 减小Batch 的大小内存
2 分析错误的位置,在哪一层出现显卡不够,好比在全链接层出现的,则下降全链接层的维度,把2048改为1042啥的tensorflow
3 增长pool 层,下降整个网络的维度。方法
4 修改输入图片的大小总结
总结如下:想进一切办法降维,下降网络的维度。错误