AssertionError: The number of GPUs ([1]) must be the same as the number of GPU indices (4) provided

Accelerated Computing Intelligent Video Analytics TAO Toolkit
15

Thanks This way it worked !!

print("To run with multigpu, please change --gpus based on the number of available GPUs in your machine.")
! tlt ssd train --gpus 4 --gpu_index 0 1 2 3 \
               -e $SPECS_DIR/ssd_train_resnet18_kitti.txt \
               -r $USER_EXPERIMENT_DIR/experiment_dir_unpruned_4GPUs\
               -k $KEY \
               -m $USER_EXPERIMENT_DIR/pretrained_resnet18/tlt_pretrained_object_detection_vresnet18/resnet_18.hdf5