1

我有一台运行两个 Intel xeon gold 6148 和 tensorflow 的服务器。当我使用 pip 安装 tf 时，我收到一条消息，指出我的安装未使用 AVX2 和 AVX512。因此，为了获得最佳性能，我尝试使用 docker 从源代码构建 tf。我是按照https://www.tensorflow.org/install/source这样做的，但是对于我使用的 bazel build 命令：

bazel build --config=mkl -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-mavx512f --copt=-mavx512pf --copt=-mavx512cd --copt=-mavx512er //tensorflow/tools/pip_package:build_pip_package

按照https://software.intel.com/en-us/articles/intel-optimization-for-tensorflow-installation-guide。
但是这个安装比标准的 pip 安装性能差很多。

所以，总结一下：安装 tensorflow 和至强金架构的最佳方式是什么？

4

1 回答 1

2

尝试使用 Intel 网站上的 tensorflow wheel。

请按照以下步骤创建您自己的 conda 环境并安装轮子。

1) conda create -n ENV_NAME -c intel python=3.6 pip numpy

2) 源激活 ENV_NAME

3）点安装https://storage.googleapis.com/intel-optimized-tensorflow/tensorflow-1.11.0-cp36-cp36m-linux_x86_64.whl

希望这可以帮助。

于 2018-12-05T05:45:47.457 回答