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
14

Could you try as below?
! tlt ssd train --gpus 2 --gpu_index 0 1 -e xxx

1 Like