Tensorflow - Resource exhausted error CPU＆GPU

Question

GPN（GTX 1060、8GB）を使用してMNISTデータセットでCNNコードをテストすると、Resource Exhausted Errorが発生します。

しかし、CPU（i7-6700、RAM：16GB）を使用して同じコードをテストすると、Resource Exhausted Errorは発生しません。

今、私の質問は、CPU & GPUの違いは何ですか？コンピューティングプロセスには何か違いがありますか？または、GPUのメモリサイズとCPUのRAMサイズの違いが原因です（実際には、CPU使用時にRAMサイズの影響を受けますか？）

Answer 1

GPUメモリが不足している可能性があります。あなたのバッチサイズを減らすか、あなたができる唯一の方法は、より大きなGPUメモリを備えたより良いコンピュータを使用することです。私のデータのbatch_sizeはコンピュータが計算するのに十分な大きさです。だから私はそれらの4つに画像を分割し、それらを別々に訓練する最終的なステップを組み合わせる。結果はそれほど悪くはありません。