Troubleshooting Guide
To share feedback or ask questions about this release, access our NVIDIA TAO Developer Forum.
Before pulling assets from NGC, ensure you run the following commands and follow the required prompts
ngc config set docker login nvcr.io
When running
ngc config set
, the NGC CLI may not prompt the user to configure the team and org. In this case, users may run into an error when downloading models sayingMissing org - If apikey is set, org is also required.
Please maintain a back-up of your existing NGC API key from the ngc config at
~/.ngc/config
and clear the ngc config by running the following command.ngc config clear
The launcher CLI abstracts the user’s interaction with the container and brings out the entrypoints inside the respective dockers.
Make sure to set your python to python3 when running the launcher. The TAO Launcher is strictly a python3 package.
When installing the TAO Launcher to your host machine’s native python3 as opposed to the recommended route of using virtual environment, you may get an error saying that
tao
binary wasn’t found. This is because the path to yourtao
binary installed by pip wasn’t added to thePATH
environment variable in your local machine. In this case, please run the following command:export PATH=$PATH:/home/$USER/.local/bin
Make sure to have all the paths required by the TAO docker to be exposed to it via the
~/.tao_mounts.json
. The launcher by default, does not have any paths mapped to it.When running the TAO Launcher, for CV applications, we recommend setting the TAO Launcher to run as the user’s host account so that the user may have permissions to edit the results directories and collaterals generated by the TAO dockers. The dockers by default are instantiated as root, so the users will need sudo access to edit the results path etc. For more information on configuring the user, please refer to the Configuring the launcher section.
When running any TAO command for the first time, the launcher pulls the container from the docker registry. This process can take a few minutes. The log will look as follows
2021-02-24 08:16:04,270 [INFO] tlt.components.docker_handler.docker_handler: The required docker doesn't exist locally/the manifest has changed. Pulling a new docker. 2021-02-24 08:16:04,270 [INFO] tlt.components.docker_handler.docker_handler: Pulling the required container. This may take several minutes if you're doing this for the first time. Please wait here.
When running tao-converter
, if you encounter the following error, ensure you are using the correct output node names (generate_detections
, mask_fcn_logits
/BiasAdd
):
[ERROR] UffParser: Output error: Output mask_head/mask_fcn_logits/BiasAdd not found
[ERROR] Failed to parse the model, please check the encoding key to make sure it's correct
The old node names in Developer Preview are generate_detections
, mask_head
/mask_fcn_logits
/BiasAdd
.
When running
tao model detectnet_v2 train ...
if you encounter errors similar to those show below, where the error shows missing variables in the checkpoint, delete the latest.ckzip
file and restart the training with the same command.Traceback (most recent call last): File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/client/session.py", line 1365, in _do_call return fn(*args) File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/client/session.py", line 1350, in _run_fn target_list, run_metadata) File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/client/session.py", line 1443, in _call_tf_sessionrun run_metadata) tensorflow.python.framework.errors_impl.NotFoundError: 2 root error(s) found. (0) Not found: Key cost_sums/cyclist-bbox not found in checkpoint [[{{node save/RestoreV2}}]] (1) Not found: Key cost_sums/cyclist-bbox not found in checkpoint [[{{node save/RestoreV2}}]] [[save/RestoreV2/_877]]
This error can be raised for the following reasons:
The checkpoint wasn’t saved properly.
The backend framework version used to generate the checkpoint was mismatched from the version used to load this checkpoint.
The experiment configuration has changed from what was stored in the checkpoint to the training graph that was initialized. For example, the checkpoint was generated in TAO 2.0 but was resumed in 3.0.
If you are starting an experiment with a different configuration (e.g. a different number of classes or training dataset size) ensure that the directory pointed to by the
-r
parameter of thetao model detectnet_v2 train
command is empty, with no previously generated checkpoints from older experiments. Resume must be used only for completing a previously launched job with the same command and same version of TAO.When resuming a DetectNet_v2 training from a checkpoint, maintain the same number of GPUs and the same command line to restart the training.