tensorflow.python.framework.errors_impl.InternalError: CUDA runtime implicit initialization on GPU:0 failed. Status: out of memory

出现这个问题是为什么？请教大佬需要怎么修改。感谢orz

感觉是显存爆了，把你的batch size搞小一些训练，再不行，简化下模型。