Невозможно выполнить DLProf из контейнера NG C TensorFlow

Question

Я хотел бы использовать утилиту командной строки dlprof. Согласно документации: https://docs.nvidia.com/deeplearning/frameworks/dlprof-user-guide/

Я установил nvidia tenorflow docker и могу запустить его docker Запустить --gpus all -it --rm -v local_dir: / root nvcr.io/nvidia/tensorflow:20.03-tf2-py3

Но я не могу ни выполнить dlprof, ни найти его. А теперь как мне это выполнить? Спасибо

Answer 1

В контейнерах NVIDIA dlprof предустановлен и может быть вызван через командную строку, как описано в quickstart . Он также показывает отличный и простой способ просмотра результатов на тензорной доске.

Для начала вы можете запустить dlprof, просто используя

dlprof [опции] [ваша команда]

Например:

dlprof --reports = summary python main.py

Ваш код будет выполнен во время профилирования от dlprof. Результат будет помещен в dlprof_summary.csv в текущем каталоге. Доступно несколько видов отчетов, таких как: сводные, подробные, ядро, тензор и др. c.

Больше вы можете найти на генерации отчетов