1. 程式人生 > >deepin下解決Tensorflow 使用時cpu編譯不支援警告

deepin下解決Tensorflow 使用時cpu編譯不支援警告

轉自https://blog.csdn.net/Fourierrr_/article/details/79749899

deepin下使用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點這裡下載。其他作業系統可以點這裡找到對應的版本。deepin Linux就可以選擇對應的Ubuntu 16.04版本的來安裝。

我用的deepin15.7,python3.5,所以選擇如下圖:

在這裡插入圖片描述

然後cd到下載目錄下:

sudo pip install tensorflow-1.11.0-cp35-cp35m-linux_x86_64.whl

執行安裝。

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