Sorry for late reply. Could you please docker pull the latest docker to run?
docker pull nvcr.io/nvidia/tao/tao-toolkit:4.0.1-tf1.15.5
Then login the docker. Similar to below.
$ docker run --runtime=nvidia -it --rm -v yourlocalfolder:/workspace nvcr.io/nvidia/tao/tao-toolkit:4.0.1-tf1.15.5 /bin/bash
Then, run command without “tao”.
# detectnet_v2 train blabla