zoukankan      html  css  js  c++  java
  • 关闭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]]
    
  • 相关阅读:
    Sizzle选择器引擎介绍
    from、includes、indexOf
    indexOf与includes的区别
    ES6 find 和 filter 的区别
    Spring MVC多个视图解析器及优先级
    Spring MVC多个视图解析器及优先级
    freemarker 中文乱码问题解决办法
    freemarker 中文乱码问题解决办法
    freemarker 中文乱码问题解决办法
    Spring集成MyBatis 通用Mapper以及 pagehelper分页插件
  • 原文地址:https://www.cnblogs.com/chay/p/10550882.html
Copyright © 2011-2022 走看看