這是tensorflow 一個常常性錯誤,錯誤的緣由在於:顯卡內存不夠。網絡
解決方法就是下降顯卡的使用內存,途徑有如下幾種措施:圖片
1 減小Batch 的大小內存
2 分析錯誤的位置,在哪一層出現顯卡不夠,好比在全鏈接層出現的,則下降全鏈接層的維度,把2048改爲1042啥的tensorflow
3 增長pool 層,下降整個網絡的維度。方法
4 修改輸入圖片的大小總結
總結如下:想進一切辦法降維,下降網絡的維度。錯誤