Trying to use my Dataset

Accelerated Computing Intelligent Video Analytics TAO Toolkit
6

So, it should be TLT 2.0 because of nvcr.io/nvidia/tlt-streamanalytics:v2.0_py3”

Please add --runtime=nvidia when you login the TLT 2.0 docker.
$ docker run --runtime=nvidia -it nvcr.io/nvidia/tlt-streamanalytics: /bin/bash

If still not work, please check the first field of a label file should be string instead of int.
See similar topic: ImportError: No module named nvml - #6 by Morganh