ImportError: No module named nvml

Accelerated Computing Intelligent Video Analytics TAO Toolkit
6

The first field of a label file should be string instead of int.
See Integrating TAO Models into DeepStream — TAO Toolkit 3.22.05 documentation