Sayed_Nadim:

to(device)


Hi Sayed,
Thank you for the reply. I am using multiple GPUs, and because it is on server, every time, I got different server. So I don’t necessarily have GPU0 to use. Also, the message shows that the input and weight are not on the same device. How does this thing happen? And what does device 1 for and why it should be 0?