「Tensorflow」错误tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.

环境

环境：docker
系统：ubuntu 18.04
显卡：rtx 1080ti
tf-v： 1.15.0

跑神经网络模型，报了标题的错误

正文

错误截图如下：

解决办法：

在入口文件的import 后加入如下代码：

config = tf.compat.v1.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.6  # 0.6 sometimes works better for folks
keras.backend.tensorflow_backend.set_session(tf.compat.v1.Session(config=config))

限制tensorflow运行内存即可