transducer-tutorial
transducer-tutorial copied to clipboard
Cuda Problem
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?
I got this error when running in Google Colab AND Local Machine