我有一台服务器,上面有两个Intel xeon gold 6148和tensorflow。当我用pip安装tf时,我会收到一条消息,即我的安装中没有使用AVX2和AVX512。因此,为了获得最好的性能,我尝试从源代码构建tf使用docker。我是在https://www.tensorflow.org/install/source之后这样做的,但是对于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黄金架构的最佳方法是什么?
发布于 2018-12-05 05:45:47
尝试使用来自英特尔网站的tensorflow砂轮。
请按照以下步骤创建您自己的conda环境并安装方向盘。
1) conda创建-n ENV_NAME -c intel python=3.6 pip numpy
2)源激活ENV_NAME
3) pip安装64.whl
希望这能有所帮助。
https://stackoverflow.com/questions/53521339
复制相似问题