No CUDA-capable device is detected

It is not related to TAO.
Please check if the hint from Nvida Container Toolkit: Failed to initialize NVML: Unknown Error - #3 by SimonBirrell can help you.