建议您添加如下代码
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
因为这可能是tf自身的bug,官方issue中有写,但目前未处理的状态,评论中提供了一些解决方法。您可以尝试加一下以上代码,看能解决该问题。
https://github.com/tensorflow/tensorflow/issues/24828
本文针对TensorFlow在GPU上运行时可能出现的内存溢出问题,提供了有效的代码配置方案。通过调整ConfigProto设置,允许GPU内存按需增长,避免了预分配所有GPU内存导致的资源浪费,有助于提高GPU使用效率。

2936

被折叠的 条评论
为什么被折叠?



