Looks nvidia-smi runs smoothly.
This is the first time to see “ImportError: No module named nvml” in TLT. Other users do not meet this issue.
How did you trigger the docker? Is it below?
docker run --runtime=nvidia -it nvcr.io/nvidia/tlt-streamanalytics:v2.0_dp_py2 /bin/bash