Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA

在執行python程序的時候,報Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA的警告。大概意思是:你的CPU支持AVX擴展,但是你安裝的TensorFlow版本無法編譯使用。

在這裏插入圖片描述

解決方法:

如果安裝的是GPU版本

如果你有一個GPU,你不應該關心AVX的支持,因爲大多數昂貴的操作將被分派到一個GPU設備上(除非明確地設置)。在這種情況下,您可以簡單地忽略此警告: 

import os 
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

如果安裝的是CPU版本(pip install tensorflow)

1.在代碼中加入如下代碼,忽略警告: 

import os 
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' 
2.編譯TensorFlow源碼 

    如果您沒有GPU並且希望儘可能多地利用CPU,那麼如果您的CPU支持AVX,AVX2和FMA,則應該從針對CPU優化的源構建tensorflow。在這個問題中已經討論過這個問題,也是這個GitHub問題。 Tensorflow使用稱爲bazel的ad-hoc構建系統,構建它並不是那麼簡單,但肯定是可行的。在此之後,不僅警告消失,tensorflow性能也應該改善。

參考博客:https://blog.csdn.net/hq86937375/article/details/79696023

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章