两行命令过滤掉tensorflow开头打印的所有红色警告

最新推荐文章于 2023-11-12 17:44:13 发布

源代码杀手

最新推荐文章于 2023-11-12 17:44:13 发布

阅读量2.1k

点赞数 4

分类专栏： TensorFlow 人工智能 ubuntu 文章标签： ubuntu

本文链接：https://blog.csdn.net/weixin_41194129/article/details/113178445

版权

ubuntu 同时被 3 个专栏收录

78 篇文章 7 订阅

订阅专栏

人工智能

77 篇文章 19 订阅

订阅专栏

TensorFlow

41 篇文章 5 订阅

订阅专栏

前提：你已经完全配置好了TF2.x环境的情况下，出现如下警告：
在这里插入图片描述
解决办法：在代码开头加上：


import os
os.environ["CUDA_VISIBLE_DEVICES"]="0"

# 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