Cuda Problem

[longlnOff]

Sir, I get this error when run your code:
Found inputs tensors to be on [device(type='cuda', index=0), device(type='cuda', index=0), device(type='cuda', index=0), device(type='cuda', index=0)] while needed to be on a 'cuda' device to use the transducer loss.

I installed cuda, torch cuda and although check it. Can you explain this error for me and show me how to fix it?

[longlnOff]

I got this error when running in Google Colab AND Local Machine