TAO crash after driver update

To run the notebook, I do the following

export PATH=$PATH:/home/pryntec/.local/bin
export WORKON_HOME=~/Envs
export VIRTUALENVWRAPPER_PYTHON=/usr/bin/python3
source $HOME/.local/bin/virtualenvwrapper.sh
workon launcher

Then,

pip show nvidia-tao

gives the new version

Name: nvidia-tao
Version: 0.1.24
Summary: NVIDIA’s Launcher for TAO Toolkit.
Home-page:
Author: Varun Praveen
Author-email: vpraveen@nvidia.com
License: NVIDIA Proprietary Software
Location: /home/pryntec/Envs/launcher/lib/python3.8/site-packages
Requires: docker, six, tabulate
Required-by:

and I have the following docker images available

REPOSITORY TAG IMAGE ID CREATED SIZE
nvcr.io/nvidia/tao/tao-toolkit-tf v3.22.05-tf1.15.5-py3 b85103564252 5 weeks ago 11.7GB
nvcr.io/nvidia/tao/tao-toolkit-tf v3.22.05-tf1.15.4-py3 ca92a571a959 5 weeks ago 16.1GB
nvcr.io/nvidia/tao/tao-toolkit-tf v3.21.11-tf1.15.4-py3 fadbda32c62f 7 months ago 16.1GB
hello-world latest feb5d9fea6a5 9 months ago 13.3kB
nvidia/cuda 11.0-base 2ec708416bb8 22 months ago 122MB

However, when I run the docker image that you suggest,

docker run --runtime=nvidia -it --rm nvcr.io/nvidia/tao/tao-toolkit-tf:v3.22.05-tf1.15.4-py3 /bin/bash

I get the following error at the end of the extraction process

chmod: cannot access ‘/opt/ngccli/ngc’: No such file or directory

which is supposed to be solved by the nvidia-tao update. I need to run

docker run --runtime=nvidia -it --rm --entrypoint “” nvcr.io/nvidia/tao/tao-toolkit-tf:v3.22.05-tf1.15.4-py3 /bin/bash

to remove the error. Is that normal ?

Moreover, in the notebook, how can I check the current docker image that is running ? (to check that tf1.15.4 is running and not tf15.5)

Thanks