在執行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