我想使用dlprof命令行实用程序。
根据文档:
https://docs.nvidia.com/deeplearning/frameworks/dlprof-user-guide/
我安装了nvidia tensorflow docker,我可以运行它
docker运行--gpus all -it --rm -v local_dir:/ root nvcr.io/nvidia/tensorflow:20.03-tf2-py3
但是我既不能执行dlprof也找不到它。
您现在如何执行它?
谢谢
最佳答案
如quickstart中所述,在NVIDIA容器中预装了dlprof,可以通过命令行调用它。它还向您展示了一种在张量板上查看结果的简便方法。
要开始使用,您可以通过简单地使用来运行dlprof
dlprof [选项] [您的命令]
例如:
dlprof --reports =摘要python main.py
您的代码将在dlprof进行配置文件时执行。结果将放入当前目录的dlprof_summary.csv中。有多种报告可用,例如:摘要,详细,内核,张量等。
您可以在report generation找到更多
关于docker - 无法从NGC TensorFlow容器执行DLProf,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/60975238/