为Intel Xeon Gold 6148构建Tensorflow

时间:2018-11-28 14:08:05

标签: tensorflow intel bazel avx2 avx512

我有一台运行两个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 xeon gold架构的最佳方法是什么?

1 个答案:

答案 0 :(得分:1)

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

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

1)康达创建-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

希望这会有所帮助。