I’m not sure if that is your problem, but using .cuda was failing for me too and I had to use the  set_device function.
Did you try it ?



 theevann:

I had to manually set the device with torch.set_device(x) . If I just use .cuda(x) , it crashes with the same cryptic error message.