Keras学习笔记(4)Resource exhausted:
2018-11-23 本文已影响0人
谢昆明
分配的显存超过GPU可用的最大显存,显存不足(OOM, Out of Memory)
分析:
这样的话可能有两个原因:
(1)batchsize太大,这种只需要将batchsize减小就行了
(2)GPU的显存太小,或者剩余的显存太少了
总共有2048M的显存,只用了481M,还剩下1500M因此显存是足够的

解决方法:
减小batchsize
目前的值是100,改为10