美文网首页TensorflowKeras学习笔记
Keras学习笔记(4)Resource exhausted:

Keras学习笔记(4)Resource exhausted:

作者: 谢昆明 | 来源:发表于2018-11-23 21:21 被阅读0次

分配的显存超过GPU可用的最大显存,显存不足(OOM, Out of Memory)

分析:
这样的话可能有两个原因:

(1)batchsize太大,这种只需要将batchsize减小就行了

(2)GPU的显存太小,或者剩余的显存太少了
总共有2048M的显存,只用了481M,还剩下1500M因此显存是足够的

image.png

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

相关文章

网友评论

    本文标题:Keras学习笔记(4)Resource exhausted:

    本文链接:https://www.haomeiwen.com/subject/hkusqqtx.html