Keras学习笔记(4)Resource exhausted: OOM when allocating tensor with shape

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

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

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

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

image.png

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

你可能感兴趣的:(Keras学习笔记(4)Resource exhausted: OOM when allocating tensor with shape)