如何静音 TensorFlow 的调试输出？

抑制 Tensorflow 调试输出

Tensorflow 在初始化阶段打印有关加载的库、找到的设备和其他调试数据的大量信息。虽然此输出对于故障排除很有价值，但它也可能令人难以承受并分散其他日志消息的注意力。本文探讨了如何禁用此调试信息。

为了抑制调试日志的显示，有一个简单而有效的解决方案。通过将环境变量 TF_CPP_MIN_LOG_LEVEL 设置为适当的值，可以控制日志记录的级别。

<code class="python">import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' 
import tensorflow as tf</code>

使用 Tensorflow 版本 0.12 和 1.0 进行测试，将环境变量设置为 3 可以有效禁用所有调试输出，只留下严重错误和警告。

为了更好地控制日志记录行为，可以设置不同的级别：

• 0：记录所有消息（默认）
• 1 ：抑制信息消息
• 2：抑制信息和警告消息
• 3：抑制信息、警告和错误消息

以上是如何静音 TensorFlow 的调试输出？的详细内容。更多信息请关注PHP中文网其他相关文章！