您的CPU 支援AVX 和AVX2:當TensorFlow 抱怨時該怎麼辦
您在使用TensorFlow 時可能遇到此警告訊息:
Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
了解警告
現代CPU 提供稱為擴充的效能增強指令,例如AVX(進階向量擴充)。 AVX 包括 FMA(融合乘法累加)運算,可顯著加快機器學習中常見的線性代數運算。此警告表示您的 CPU 確實支援 AVX,但 TensorFlow 未配置為使用它。
為什麼預設不使用 AVX?
TensorFlow 的預設分佈是建置時不支援這些擴展,以確保與各種 CPU 的兼容性。此外,在機器學習訓練方面,GPU 的效能通常優於 CPU,因此預設建置專注於 GPU 相容性。
解決問題
使用GPU:
如果您有GPU,TensorFlow 將自動優先處理計算操作密集型,從而降低CPU 的AVX 支援的相關性。若要抑制警告,請設定:
import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
沒有GPU:
要充分利用CPU 的潛力,請考慮使用AVX、AVX2、如果您的CPU 支援,則啟用FMA。雖然建置過程很複雜(涉及 bazel 建置系統),但它應該消除警告並增強 TensorFlow 在 CPU 上的效能。
以上是TensorFlow AVX/AVX2 警告:如何利用 CPU 指令來獲得更好的效能?的詳細內容。更多資訊請關注PHP中文網其他相關文章!