首頁  >  文章  >  後端開發  >  如何靜音 TensorFlow 偵錯訊息?

如何靜音 TensorFlow 偵錯訊息?

Linda Hamilton
Linda Hamilton原創
2024-11-01 13:05:29649瀏覽

How to Silence TensorFlow Debugging Messages?

如何抑制 TensorFlow 偵錯資訊

TensorFlow 使用者可能會遇到有關載入的庫和可用裝置的無關控制台訊息。這些訊息可能具有侵入性,尤其是在生產環境或偵錯 Python 程式碼時。

解決方案:停用偵錯訊息

要停用此偵錯訊息,請設定TF_CPP_MIN_LOG_LEVEL 環境變數:

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

說明:

TF_CPP_MIN_LOG_LEVEL 環境變數控制列印訊息的最低嚴重程度。透過將其設為“3”,所有嚴重程度低於 3 的訊息(即訊息、警告和錯誤)都會被抑制。

日誌層級層次結構:

  • 0:列印所有訊息(預設行為)
  • 1:不列印INFO 訊息
  • 2:不列印INFO 和WARNING 訊息
  • 3:不會列印訊息、警告和錯誤訊息

此解決方案已測試並確認可與TensorFlow 版本0.12和1.0 配合使用。

以上是如何靜音 TensorFlow 偵錯訊息?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn