Tensorflow 使用時cpu編譯報錯your CPU supports instructions that this TensorFlow binary was not compiled to

使用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.忽略這個警告,不看它!

1

2

3

4

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

1

pip uninstall tensorflow

完成後可以用pip list查看一下所有庫,檢查是否成功卸載。

然後去github下載正確的tf版本,Windows點這裏下載。其他操作系統可以點這裏找到對應的版本。

博主用的win10,python3.6,所以選擇如下圖:

對應的path在上邊的找到對應的.whl下載即可。

然後安裝:

1

pip install tensorflow-1.6.0-cp36-cp36m-win_amd64.whl

成功安裝後,再次運行TensorFlow就不會報錯啦~

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