【问题标题】:Performance Analysis tools for TensorflowTensorFlow 的性能分析工具
【发布时间】:2023-09-26 23:01:01
【问题描述】:

我正在尝试使用 Tensorflow、CNTK 和 MxNet 运行一些机器学习算法。这些框架是否有一些特定的性能分析工具,特别是用于分析运行时信息(内存、通信等)?

【问题讨论】:

    标签: tensorflow cntk mxnet


    【解决方案1】:

    仅供参考:MXNet 有一个内置的分析器。您可以在此处找到有关它的更多信息:

    https://mxnet.incubator.apache.org/how_to/perf.html#profiler

    【讨论】:

      【解决方案2】:

      只能回答有关 Tensorflow 的问题。它具有嵌入式跟踪,但它只能为您提供一些基本的性能特征。但是,有a way how you can import 其跟踪数据到 Intel VTune Amplifier(该工具不是免费的,但有免费的全功能学术版和试用版)并获得更高级的指标(各种块的利用率、它们之间的带宽、IO 等) 映射到张量。

      【讨论】: