如何使用TensorFlow对两个Anaconda环境进行基准测试?

时间:2018-07-21 17:48:38

标签: python-3.x tensorflow

我已经厌倦了所有关于TensorFlow尚未针对我的CPU优化的提醒,因此最终从源代码对其进行了编译。实际上,我做了两次,制作了两个.whl文件,一次使用

bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package

并使用一次

bazel build --config=mkl --config=opt //tensorflow/tools/pip_package:build_pip_package

因为我不确定Intel MKL会有什么不同。现在,我设置了两个相同的anaconda环境,一个使用每个whl文件。

确定哪种.whl软件包在我的系统上性能更好的最快方法是什么?如果有人可以将我指向tensorflow中的标准Benchmark软件包/命令,那将很棒(请注意,我没有GPU支持)。

1 个答案:

答案 0 :(得分:2)

您可以运行以下基准测试并检查性能。

https://github.com/tensorflow/benchmarks.git

git将上述代码克隆到您的终端,然后运行tf_cnn_benchmark.py基准代码。

谢谢