如何阻止TensorFlow打印大量cudart64相关的警告信息？

解决方法 #

你遇到的cudart64相关警告是TensorFlow检测到设备支持CUDA但未安装CUDA运行库触发的，本身不影响CPU模式下的模型运行，可通过以下方法屏蔽警告：

• 方法1：修改代码（最常用）
  在代码最开头、导入TensorFlow之前添加环境变量配置即可：

import os
# 日志等级设置为2，仅输出ERROR及以上级别的信息，自动屏蔽INFO、WARNING类通知
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import tensorflow as tf

如果仍有少量CUDA相关警告残留，可以在导入TensorFlow后额外加一行配置：

tf.get_logger().setLevel('ERROR')

• 方法2：无需修改代码的全局设置
  运行Python脚本前先在终端配置临时环境变量即可：
  • Windows cmd：执行 set TF_CPP_MIN_LOG_LEVEL=2 后再运行你的脚本
  • Linux/macOS终端：执行 export TF_CPP_MIN_LOG_LEVEL=2 后再运行你的脚本

以上配置都不会影响你现有图像分类器的正常功能，仅会屏蔽非必要的警告输出。

内容的提问来源于stack exchange，提问作者Aadhi Arun