tensorflow keras 关闭打印warning

skyecs

于 2023-06-30 15:15:03 发布

阅读量664

点赞数

CC 4.0 BY-SA版权

文章标签： tensorflow keras 人工智能

本文链接：https://blog.youkuaiyun.com/skyecs/article/details/131477266

通过设置环境变量TF_CPP_MIN_LOG_LEVEL为3，并使用tf.get_logger().setLevel(ERROR)，可以确保在TensorFlow2.3中屏蔽所有警告信息，只显示错误级别日志。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用下面代码设置TensorFlow的日志级别为ERROR，以及将环境变量TF_CPP_MIN_LOG_LEVEL设置为'3'，这将禁用TensorFlow的所有警告信息。os.environ部分一定要放在import tensorflow之前，我用的是tensorflow2.3，不加tf.get_logger().setLevel('ERROR')也会不生效。

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

import tensorflow as tf
tf.get_logger().setLevel('ERROR')

import os
os.environ["TF_CPP_MIN_LOG_LEVEL"]='1'
# 默认的显示等级，显示所有信息
os.environ["TF_CPP_MIN_LOG_LEVEL"]='2'
# 只显示 warning 和 Error  
os.environ["TF_CPP_MIN_LOG_LEVEL"]='3'
# 只显示 Error