我有一台运行两个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架构的最佳方法是什么?
答案 0 :(得分:1)
尝试使用Intel网站上的tensorflow wheel。
请按照以下步骤创建自己的conda环境并安装滚轮。
1)康达创建-n ENV_NAME -c intel python = 3.6 pip numpy
2)源激活ENV_NAME
希望这会有所帮助。