禁止tensorflow提示warning

TensorFlow的日志级别分为以下三种:

TF_CPP_MIN_LOG_LEVEL = 1         //默认设置,为显示所有信息                
TF_CPP_MIN_LOG_LEVEL = 2         //只显示error和warining信息                
TF_CPP_MIN_LOG_LEVEL = 3         //只显示error信息
所以,当TensorFlow出现警告信息,又不想让警告信息显示时,可进行如下设置:

方法一
python环境下通过在python文件中添加如下两行代码,设置TensorFlow日志输出级别。
import os
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"

方法二
在linux环境下,我们可以通过linux命令,屏蔽掉TensorFlow的warning:
1、暂时屏蔽warning:
在linux终端中输入命令:export TF_CPP_MIN_LOG_LEVEL=2,当该终端关闭,既失效;
2、永久屏蔽warning:
修改~/.bashrc文件,

gedit ~/.bashrc

在该文件中加入:
export TF_CPP_MIN_LOG_LEVEL=2
但是修改后需要重启系统或者使用source ~/.bashrc命令,使修改后的文件生效。

你可能感兴趣的:(禁止tensorflow提示warning)