TAO v5.5.0
NVIDIA TAO v5.5.0

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

    Copy
    Copied!
                

    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 saying

    Copy
    Copied!
                

    Missing 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.

    Copy
    Copied!
                

    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 your tao binary installed by pip wasn’t added to the PATH environment variable in your local machine. In this case, please run the following command:

    Copy
    Copied!
                

    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

    Copy
    Copied!
                

    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):

Copy
Copied!
            

[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.

    Copy
    Copied!
                

    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 the tao 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.

Previous Frequently Asked Questions
Next Support Information
© Copyright 2024, NVIDIA. Last updated on Aug 30, 2024.