為什麼 TensorFlow 會發出關於不支援的 AVX/AVX2 指令的警告，以及如何修復它？

“您的CPU 支持此TensorFlow 二進製文件未編譯為使用的指令：AVX AVX2”

問題：

我安裝了TensorFlow 並收到一條警告訊息，指出我的CPU支援二進位檔案未編譯使用的指令。為什麼會發生這種情況，我該如何解決？

答案：

警告是什麼？

AVX（高階向量擴展） 和FMA（融合乘加） 是x86 指令集架構的擴展，提供更快的線性代數計算，如點積、矩陣乘法和卷積。現代 CPU 支援這些擴展，可以顯著加快機器學習訓練速度。

那為什麼不使用它？

預設的 TensorFlow 發行版旨在與盡可能多的 CPU 相容。它旨在與 GPU 一起使用，在執行大規模訓練任務時，GPU 的速度比 CPU 快得多。

要做什麼：

如果您有 GPU：

• 使用 os.environ['TF_CPP_MIN_LOG_LEVEL'] 忽略警告'2'.

如果您沒有GPU 且想要使用CPU：

• 從原始碼建置TensorFlow

從原始碼建置TensorFlow 啟用🎜> 啟用AVX、AVX2 和FMA。這需要使用 Bazel 建置系統，如連結討論中所述。這應該消除警告並提高性能。

以上是為什麼 TensorFlow 會發出關於不支援的 AVX/AVX2 指令的警告，以及如何修復它？的詳細內容。更多資訊請關注PHP中文網其他相關文章！