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

问题描述 投票:0回答:1

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

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

我的问题是:

  1. 我该如何解决?特别是,我希望能够保留当前的TensorFlow版本(1.12.0)
  2. 考虑到我使用GPU,我会获得可观的收益吗?

我使用Ubuntu 18.04.1 LTS。

谢谢;)

tensorflow ubuntu installation cpu
1个回答
0
投票
  1. 我不知道如何保留1.12.0,但是Tensorflow页面有一个很好的构建指南:https://www.tensorflow.org/install/source#setup_for_linux_and_macos
© www.soinside.com 2019 - 2024. All rights reserved.