标签： memory-management tensorflow gpu
我正在做教程 TF图层指南：构建卷积神经网络并得到了
资源耗尽：OOM在分配形状[10000,32,28,28] 的张量时
资源耗尽：OOM在分配形状[10000,32,28,28]
在评估阶段。由于我的GPU是只有2GB内存的GeForce GTX 680，所以在玩批量大小后我并不感到惊讶，我对GPU内存的管理方式有点困惑。
我能够使用批量大小2500来评估模型，但是我对使用了多少内存以及TF / python / numpy如何管理它感到困惑。
我选择了一批2750我无法评估模型..我怎样才能估计出我需要多少内存？