TAO Toolkit 4.0 Google Colab training error: No module "wandb"

When I run this Google Colab YOLO 3 workbook training script:

print(“To run with multigpu, please change --gpus based on the number of available GPUs in your machine.”)
!tao yolo_v3 train -e $SPECS_DIR/yolo_v3_train_resnet18_tfrecord.txt
-r $EXPERIMENT_DIR/experiment_dir_unpruned
-k $KEY
–gpus 1

I get the error :
ModuleNotFoundError: No Module named “wandb”

To run with multigpu, please change --gpus based on the number of available GPUs in your machine.
Using TensorFlow backend.
2023-01-23 01:51:05.020090: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
WARNING:tensorflow:Deprecation warnings have been disabled. Set TF_ENABLE_DEPRECATION_WARNINGS=1 to re-enable them.
Traceback (most recent call last):
File “/usr/local/bin/yolo_v3”, line 8, in
File “”, line 12, in main
File “”, line 266, in launch_job
File “”, line 50, in get_modules
File “/usr/lib/python3.6/importlib/init.py”, line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File “”, line 994, in _gcd_import
File “”, line 971, in _find_and_load
File “”, line 955, in _find_and_load_unlocked
File “”, line 665, in _load_unlocked
File “”, line 678, in exec_module
File “”, line 219, in _call_with_frames_removed
File “</usr/local/lib/python3.6/dist-packages/iva/yolo_v3/scripts/train.py>”, line 3, in
File “”, line 18, in
File “</usr/local/lib/python3.6/dist-packages/iva/yolo_v3/models/utils.py>”, line 1, in
File “”, line 4, in
File “</usr/local/lib/python3.6/dist-packages/iva/common/mlops/wandb.py>”, line 1, in
File “”, line 9, in
ModuleNotFoundError: No module named ‘wandb’

I have run the
2.3 Setup Python environment
script earlier in the Google colab workbook

As a workaround you can add !pip install wandb before the tao command. We’ll check and come back on the root cause.

There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks

We fix the issue. Please retry. Thanks.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.