解决TensorFlow CPU编译不支持AVX/AVX2警告

TensorFlow CPU AVX警告现象与原因

在使用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版本，而非针对特定CPU指令集优化的版本。

解决TensorFlow CPU AVX警告的两种方案

针对此问题，主要有两种解决思路：

第一种方法是忽略该警告信息，不进行任何额外操作。

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

对于大多数用户而言，采用第二种方法更为推荐。

方案二：更换支持AVX2的TensorFlow版本

第二种方法是彻底解决，即更换为支持CPU使用AVX2指令集编译的TensorFlow版本。

首先，需要卸载当前版本的TensorFlow。

pip uninstall tensorflow

卸载完成后，可以使用pip list命令查看所有已安装的库，以确认TensorFlow是否已成功移除。

接着，前往GitHub下载正确的TensorFlow版本。Windows用户可点击此处下载，其他操作系统用户可点击此处查找对应版本。

以博主使用的Win10系统和Python 3.6环境为例，选择如下图所示的对应版本：

在页面中找到对应的.whl文件并下载即可。

随后执行安装命令：

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

成功安装后，再次运行TensorFlow程序，该警告便不再出现。

以上即为解决TensorFlow使用时CPU编译不支持警告的完整内容，希望能对读者有所帮助。