DLProf Viewer

Install using Python Wheel

Create virtual environment

python -m venv venv
source venv/bin/activate

Install required packages

pip install setuptools pip wheel

pip install nvidia-pyindex

pip install nvidia-dlprof

TensorFlow Containers

Tags

docker pull nvcr.io/nvidia/tensorflow:<25.02>-tf2-py3

docker run --rm --gpus=1 --shm-size=1g --ulimit memlock=-1 \
--ulimit stack=67108864 -it -p8000:8000 -v/full/path/to/training/data:/data \
nvcr.io/nvidia/tensorflow:25.02-tf2-py3

dlprof python <training-script.py>