Refer to Train with my own tlt model #2 - #17 by tmid0819 and No CUDA-capable device is detected on tao detectnet_v2 dataset convert - #4 by NilsAI