使用TensorFlow后端隐藏。 TensorFlow 2.0 Alpha中的消息

时间:2019-03-19 01:03:46

标签: python tensorflow keras

导入TensorFlow 2.0 Alpha时,将显示以下消息:

Using TensorFlow backend.

如何隐藏此消息,使其无法打印出来?

操作系统是Windows,我尝试了以下操作:

import os
import sys
stderr = sys.stderr
sys.stderr = open(os.devnull, 'w')
import tensorflow as tf
sys.stderr = stderr

不能消除该消息。

TensorFlow版本是

print(tf.__version__)
2.0.0-alpha0

此打印输出似乎来自Keras:

https://github.com/keras-team/keras/issues/1406

Keras disable showing using XXX backend

1 个答案:

答案 0 :(得分:0)

我有一个linux系统,所以我不确定这是否对您有用,但是以下代码禁止显示该消息:

import logging
logger = tf.get_logger()
logger.setLevel(logging.ERROR)

您应该只看到错误消息的打印输出。 (来自https://github.com/tensorflow/tensorflow/issues/26662