Question

我使用提供的默认指令here在ubuntu 16.04上成功安装了cpu only tensorflow。建议使用virtualenv和pip的指令，所以我没有从源代码构建。按照这些说明安装我没有任何问题。

我使用同一页面上提供的further down说明验证了我的安装，并且在程序成功运行时，它会输出以下警告。

W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE3 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computations.
Hello, TensorFlow!

我查看的所有stackoverflow页面只针对使用bazel从源代码构建的人，但似乎并不适用于使用pip的人。

如何重新编译或更新我的安装以获取SSE说明？

Answer 1

您必须使用bazel和--config=opt选项构建为您的体系结构自定义的轮子，然后使用pip安装生成的轮子。

我计划偶尔将2014 MacBook + Xeon V3优化版本上传到https://github.com/yaroslavvb/tensorflow-community-wheels

安装Bazel后，为pip命令制作一个轮子大致如下

./configure

export LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH"
export flags="--config=opt --config=cuda -k"

bazel build $flags -k //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

然后将轮子放入/tmp/tensorflow_pkg

Answer 2

bazel build --linkopt='-lrt' -c opt --copt=-mavx --copt=-msse4.2 --copt=-msse4.1 --copt=-msse3-k //tensorflow/tools/pip_package:build_pip_package

如果上述问题已解决，请在How do I resolve these tensorflow warnings?

上投票Christian Frei's

Answer 3

这些只是警告。根据此link，在导入TensorFlow之前添加此行，

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

我也有同样的问题，但建议的链接解决了我的问题。

另外看一下这个answer，它将有助于更多地了解SSE信息以及不同类型的日志指示器。

使用带有pip的SSE指令进行Tensorflow安装

3 个答案: