I think your latest comment is not related to this topic.
For original error, please check if you run below before running “pip install nvidia-tao==4.0.1”
$ pip3 install nvidia-pyindex
After I ran pip install nvidia-tao ==4.0.1, I ran bash quickstart_launcher.sh upgrade and pip3 install nvidia-tao --upgrade. Now the module error messages disappear.