停止 Tensorflow 打印到控制台

Question

我一直使用tensorflow没有问题，直到我添加了以下代码行：log_dir = os.path.join("logs",                       "fit",                       datetime.datetime.now().strftime("%Y%m%d-%H%M%S"))tensorboard_callback = TensorBoard(log_dir)运行此命令后，我会在控制台上打印大量信息。我尝试查看 tf.keras.callbacks.TensorBoard 文档，看看是否可以减少冗长，但我没有看到任何选项。从各种stackoverflow 答案中，我也尝试设置tfdown 的详细程度，但无济于事：tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)tf.get_logger().setLevel('ERROR')tf.autograph.set_verbosity(3)os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'我有以下规格：Python = 3.8Tensorflow = 2.3.1Cuda Toolkit = 10.1 cuDNN = 7.6.4GPU=Nvidia RTX2060打印到控制台的信息都是I消息

Answer 1

您可以使用 禁用调试日志os.environ。

import os

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' 

import tensorflow as tf

可能的值如下：

0 = all messages are logged (default behavior)

1 = INFO messages are not printed

2 = INFO and WARNING messages are not printed

3 = INFO, WARNING, and ERROR messages are not printed