如何控制终端中的 TensorFlow 调试输出？

在终端中控制 TensorFlow 调试信息

使用 TensorFlow 时，终端中会显示各种调试信息，包括有关加载的库和内容的详细信息检测到的设备。虽然此信息对于调试很有用，但它可能会变得令人不知所措或分散注意力。为了解决这个问题，TensorFlow 提供了一种自定义记录调试信息级别的机制。

禁用调试信息

要禁用所有调试信息，请将 TF_CPP_MIN_LOG_LEVEL 环境变量设置为3. 这将抑制来自 TensorFlow 的所有信息性消息。

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

日志级别自定义

TF_CPP_MIN_LOG_LEVEL 变量允许对日志记录级别进行更精细的控制。以下值代表不同的日志记录级别：

• 0：记录所有消息（默认）。
• 1：不打印 INFO 消息。
• 2：INFO和 WARNING 消息不会被打印。
• 3：不会打印 INFO、WARNING 和 ERROR 消息。

示例

以下示例演示了如何抑制所有非错误消息：

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

测试版本

此解决方案已在 TensorFlow 版本 0.12 和 1.0 上进行了测试。

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