Tensorflow - 资源耗尽错误CPU＆amp; GPU

Question

当我用MNIST数据集测试CNN代码时，使用GPU（GTX 1060,8GB）会出现资源耗尽错误。

但是，当我测试相同的代码时，使用CPU（i7-6700，RAM：16GB），没有资源耗尽错误。

现在，我的问题是，CPU与CPU之间的区别是什么？ GPU？它在计算过程中有什么不同之处？或者仅仅因为GPU内存大小和CPU RAM大小不同？（实际上，我不确定何时使用CPU，是否受RAM大小的影响？）

Answer 1

也许你的GPU内存不够，你可以尝试减少你的批量大小，或者你可以做的唯一方法是使用更好的GPU内存更大的计算机。我有同样的问题但是我的数据的一个batch_size足够大电脑计算。所以我将图像分成4个并分别训练它们将它们组合成最后一步。结果并不差。