时间戳

首页 > 解决方案 > 无法从 NGC TensorFlow 容器执行 DLProf

docker - 无法从 NGC TensorFlow 容器执行 DLProf

问题描述

我想使用 dlprof 命令行实用程序。根据文档： https ://docs.nvidia.com/deeplearning/frameworks/dlprof-user-guide/

我安装了 nvidia tensorflow docker，我可以运行它 docker run --gpus all -it --rm -v local_dir:/root nvcr.io/nvidia/tensorflow:20.03-tf2-py3

但我既不能执行 dlprof 也找不到它。你现在我该如何执行它？谢谢

标签： dockernvidiaprofiler

解决方案

在 NVIDIA 容器中，dlprof 是预安装的，可以通过命令行调用，如quickstart中所述。它还向您展示了一种在 tensorboard 中查看结果的简便方法。

为了让您开始，您可以通过简单地使用来运行 dlprof

dlprof [选项] [你的命令]

例如：

dlprof --reports=summary python main.py

您的代码将在 dlprof 分析时执行。结果将被放入当前目录的 dlprof_summary.csv 中。有多种报告可供选择，例如：摘要、详细、内核、张量等。

您可以在报告生成中找到更多信息

推荐阅读