关闭tensorflow运行时的警告信息

2024年3月3日 324次阅读 来源: tensorflow

执行简单的矩阵相乘的程序：

import tensorflow as tf


m1 = tf.constant([[3,3]])
m2 = tf.constant([[2],[3]])

product = tf.matmul(m1,m2)

print(product)

sess = tf.Session()

result = sess.run(product)
print(result)

sess.close()

输出：

Tensor("MatMul:0", shape=(1, 1), dtype=int32)
2019-03-18 10:44:35.419435: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
2019-03-18 10:44:35.797977: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1432] Found device 0 with properties: 
name: GeForce GTX 1050 major: 6 minor: 1 memoryClockRate(GHz): 1.493
pciBusID: 0000:01:00.0
totalMemory: 2.00GiB freeMemory: 1.61GiB
2019-03-18 10:44:35.798454: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1511] Adding visible gpu devices: 0
2019-03-18 10:44:37.076586: I tensorflow/core/common_runtime/gpu/gpu_device.cc:982] Device interconnect StreamExecutor with strength 1 edge matrix:
2019-03-18 10:44:37.076836: I tensorflow/core/common_runtime/gpu/gpu_device.cc:988]      0 
2019-03-18 10:44:37.076991: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1001] 0:   N 
2019-03-18 10:44:37.077308: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 1353 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1050, pci bus id: 0000:01:00.0, compute capability: 6.1)
[[15]]

为了避免输出上面的警告信息，在程序开头加入：

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
#默认为0：输出所有log信息
#设置为1：进一步屏蔽INFO信息
#设置为2：进一步屏蔽WARNING信息
#设置为3：进一步屏蔽ERROR信息

此时再执行上述的程序，输出：

Tensor("MatMul:0", shape=(1, 1), dtype=int32)
[[15]]
    原文作者：tensorflow
    原文地址: https://www.cnblogs.com/chay/p/10550882.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞