Model training using Tao toolkit with colab not working anymore

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

Hi Team,
I am unable to run any of the google colab notebooks using Toa Toolkit. Has the support discontinued to the notebooks.

!tao model yolo_v4_tiny dataset_convert -d $SPECS_DIR/yolo_v4_tiny_tfrecords_kitti_train.txt \ -o $DATA_DIR/train/tfrecords/train

/bin/sh: 1: yolo_v4_tiny: not found

i keep on getting the same error for all the models like ssd, yolo_v4. I have tried different gpus but they all end up giving me the same error. Can someone check if the google colab version is having some dependency issue with the colab code.
None of the notebooks mentioned below seem to be working:Running TAO on Google Colab - NVIDIA Docs

3

Usually it is because TAO is not installed successfully. Could you share the colab notebook?

4

Here is the link to the colab. I tend to get the same error with all the colabs provided in Running TAO on Google Colab - NVIDIA Docs

5

There are lots of errors when run into the cell.

import os
if os.environ["GOOGLE_COLAB"] == "1":
    os.environ["bash_script"] = "setup_env.sh"
else:
    os.environ["bash_script"] = "setup_env_desktop.sh"

os.environ["NV_TAO_TF_TOP"] = "/tmp/tao_tensorflow1_backend/"

!sed -i "s|PATH_TO_TRT|$trt_untar_folder_path|g" $COLAB_NOTEBOOKS_PATH/tensorflow/$bash_script
!sed -i "s|TRT_VERSION|$trt_version|g" $COLAB_NOTEBOOKS_PATH/tensorflow/$bash_script
!sed -i "s|PATH_TO_COLAB_NOTEBOOKS|$COLAB_NOTEBOOKS_PATH|g" $COLAB_NOTEBOOKS_PATH/tensorflow/$bash_script

!sh $COLAB_NOTEBOOKS_PATH/tensorflow/$bash_script

For example,

Traceback (most recent call last):
  File "release/docker/build_kernels.py", line 7, in <module>
    from nvidia_tao_tf1.core.utils.path_utils import expand_path
  File "/tmp/tao_tensorflow1_backend/nvidia_tao_tf1/core/__init__.py", line 18, in <module>
    from nvidia_tao_tf1.core import distribution
  File "/tmp/tao_tensorflow1_backend/nvidia_tao_tf1/core/distribution/__init__.py", line 20, in <module>
    from nvidia_tao_tf1.core.distribution.distribution import Distributor, HorovodDistributor, hvd
  File "/tmp/tao_tensorflow1_backend/nvidia_tao_tf1/core/distribution/distribution.py", line 24, in <module>
    import tensorflow as tf
ModuleNotFoundError: No module named 'tensorflow'

Need to check further. The script is in nvidia-tao/tensorflow/setup_env.sh at main · NVIDIA-AI-IOT/nvidia-tao · GitHub.

6

Suggest to use the dgpu or cloud to run.