TensorflowKeras学习笔记

Keras学习笔记(4)Resource exhausted:

2018-11-23  本文已影响0人  谢昆明

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

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

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

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

image.png

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

上一篇 下一篇

猜你喜欢

热点阅读