Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2

最新推荐文章于 2023-08-09 10:12:08 发布

Buzz.Lightyear

最新推荐文章于 2023-08-09 10:12:08 发布

阅读量636

点赞数

分类专栏： TensorFlow 文章标签： TensorFlow 安装 AVX2

本文链接：https://blog.csdn.net/weixin_43866211/article/details/104269594

版权

TensorFlow 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

TensorFlow installation and addition

pip install tensorflow==1.8 (实际中基本以1.8和1.9为主）最新的是2.1版本

安装是终端警告如下

原因：
除了通常的算术和逻辑，现代CPU提供了许多低级指令，称为扩展，例如， SSE2，SSE4，AVX等来自维基百科：

高级矢量扩展（AVX）是英特尔在2008年3月提出的英特尔和AMD微处理器的x86指令集体系结构的扩展，英特尔首先通过Sandy Bridge处理器在2011年第一季度推出，随后由AMD推出Bulldozer处理器在2011年第三季度.AVX提供了新功能，新指令和新编码方案。
特别是，AVX引入了融合乘法累加（FMA）操作，加速了线性代数计算，即点积，矩阵乘法，卷积等。几乎所有机器学习训练都涉及大量这些操作，因此将会支持AVX和FMA的CPU（最高达300％）更快。该警告指出您的CPU确实支持AVX（hooray！）。

在此强调一下：

这只限于CPU。如果你有一个GPU，你不应该关心AVX的支持，因为大多数昂贵的操作将被分派到一个GPU设备上（除非明确地设置）。在这种情况下，您可以简单地忽略此警告。

那为什么会出现这种警告呢？

由于tensorflow默认分布是在没有CPU扩展的情况下构建的，例如SSE4.1，SSE4.2，AVX，AVX2，FMA等。默认版本（来自pip install tensorflow的版本）旨在与尽可能多的CPU兼容。另一个观点是，即使使用这些扩展名，CPU的速度也要比GPU慢很多，并且期望在GPU上执行中型和大型机器学习培训。

解决方法：
如果安装的是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'

原因：
除了通常的算术和逻辑，现代CPU提供了许多低级指令，称为扩展，例如， SSE2，SSE4，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性能也应该改善。

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import tensorflow as t

a = t.constant(4)
b = t.constant(2)
c = t.add(a,b)
with t.Session() as sess:
    c = sess.run(c)
    print(c)

# output
 6