报错:Allocator (GPU_0_bfc) ran out of memory trying to allocate 3.02GiB with freed_by_count=0......
报错原因在于GPU显存不够用了。
解决方法:尝试用上单个GPU的所有显存(如果代码没有设置允许tensorflow根据自动增加申请的显存)
代码如下:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
Reference: