3 Comments

Hello thank you for all urs tutos, however i don't know if u could help me i m getting this erreor RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument mat2 in method wrapper_CUDA_mm) when running ur notebook !

Expand full comment

Is it happening when you test the model for inference?

Are you running it on Google Colab? If yes, which GPU are you using?

Expand full comment

yes sir using google colab, However i don't know how but now it's working

Expand full comment