PyTorch Training profile with nvprof and NVTX.
To simplify the working environment configuration, the profile commands depend on NGC PyTorch container. If your working environment is not ready to use NGC, please visit https://ngc.nvidia.com and configure your working environment following the NGC user guide.