關閉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]]


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM