解決Tensorflow 使用時cpu編譯不支持警告的問題
使用TensorFlow模塊時,彈出錯誤Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
原因是下載TensorFlow的版本不支持cpu的AVX2編譯。
可能是因為安裝時使用的pip install tensorflow ,這樣默認會下載X86_64的SIMD版本。
有兩種解決辦法:
1.忽略這個警告,不看它!
import os os.environ["TF_CPP_MIN_LOG_LEVEL"]='1' # 這是默認的顯示等級,顯示所有信息 os.environ["TF_CPP_MIN_LOG_LEVEL"]='2' # 只顯示 warning 和 Error os.environ["TF_CPP_MIN_LOG_LEVEL"]='3' # 只顯示 Error
我們用第二個就可以了。
2.徹底解決,換成支持cpu用AVX2編譯的TensorFlow版本。
首先,卸載原來版本的TensorFlow
pip uninstall tensorflow
完成后可以用pip list查看一下所有庫,檢查是否成功卸載。
然后去github下載正確的tf版本,Windows點這里下載。其他操作系統(tǒng)可以點這里找到對應(yīng)的版本。
博主用的win10,python3.6,所以選擇如下圖:
對應(yīng)的path在上邊的找到對應(yīng)的.whl下載即可。
然后安裝:
pip install tensorflow-1.6.0-cp36-cp36m-win_amd64.whl
成功安裝后,再次運行TensorFlow就不會報錯啦~
以上這篇解決Tensorflow 使用時cpu編譯不支持警告的問題就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。