首页 > 解决方案 > 安装 TensorFlow Profiler for Dummies

问题描述

我正在尝试按照本指南安装 TensorFlow Profiler,以更好地理解为什么我最近安装的 Keras 确实在 GPU 上运行,但几乎不使用任何资源(而且非常慢)。但是,我无法得出任何结果,因为该指南没有为我提供足够的信息,因为我不是专业的程序员并且显然缺乏必要的知识。

到目前为止我尝试了什么?我使用 Anaconda 并安装了运行版本的 python 3.7。我还安装了 tensorflow 和必要的驱动程序等,以便 tensorflow 能够访问我的 GPU。按照链接指南,我下载了“install_and_run.py”并尝试使用 conda 提示符执行它。我被要求指定--envdir 和--logdir。我在哪里指出这些?环境目录只是我当前 conda 环境的目录吗?由于我尝试将 envdir 和 logdir 都指向该方向并最终出现命令错误

True”是未知的,“True”返回非零退出状态 1。

我想不出任何解决方案。可能应该提到的是,我在使用 conda 提示符运行 .py-files 方面经验很少,通常只使用它来安装包。

我也不确定讨论 CUPTI 路径的后续步骤是什么意思。据我所知,给定的路径不是完整的路径。我应该在哪里寻找它?还是我打算执行其中的一些

/sbin/ldconfig -N -v $(sed 's/:/ /g' <<< $LD_LIBRARY_PATH) | \
grep libcupti

作为命令?我试过运行/sbin/ldconfig -N -v $,但我的系统找不到路径(可能是因为我从错误的目录开始查找?)。

任何帮助深表感谢。很抱歉来自一个困惑的人的潜在混乱帖子。

谢谢!

标签: pythontensorflowkerasanacondatensorboard

解决方案


Tensorflow 分析器不再与 Tensorboard 捆绑在一起。在拟合 Keras 模型时,有一个关于如何安装和运行它的教程。

总结是:

  1. 在你的环境中运行pip install tensorboard_plugin_profile
  2. 像往常一样声明一个张量板回调
tboard_callback = tf.keras.callbacks.TensorBoard(log_dir = logs,
                                                 histogram_freq = 1,
                                                 profile_batch = '500,520')
  1. 适合您的模型(使用声明的张量板回调)
  2. 在单独的终端(激活您的环境)上运行tensorboard --logdir=path/to/logs

教程中显示的 Profiler 选项卡可能不可见,但右上角的下拉菜单中应该有一个可用的配置文件选项。


推荐阅读