如何编译 TensorFlow 二进制文件以使用 AVX2、AVX512F、FMA？

Posted 2023-02-16

【中文标题】如何编译 TensorFlow 二进制文件以使用 AVX2、AVX512F、FMA？

【英文标题】：How to compile TensorFlow binary to use AVX2, AVX512F, FMA?

【发布时间】：2020-02-29 11:37:34

【问题描述】：

当我在 TensorFlow 中调用 tf.Session() 时，我收到以下警告消息：

I tensorflow/core/platform/cpu_feature_guard.cc:141] 你的 CPU 支持 此 TensorFlow 二进制文件未编译使用的指令：AVX2 AVX512F FMA

我的问题是：

我该如何解决这个问题？特别是，我希望能够保留当前的 ​​TensorFlow 版本（1.12.0） 考虑到我在 GPU 上工作，我会获得可观的收益吗？

我使用 Ubuntu 18.04.1 LTS。

谢谢;)

【参考方案1】：

我不知道如何保留 1.12.0，但是 Tensorflow 页面有一个很好的构建指南：https://www.tensorflow.org/install/source#setup_for_linux_and_macos

根据comments 来自Tensorflow github 项目的thread，没有。引用：

从我的实验中，我发现 CPU 优化的 GPU TF 不会显着提升性能，但它可以让 CPU 变冷。​​