cuda error

[arkinrc]

Hi,
can not use cuda but can use cpu for train. error is given below;

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

regards
arkin

[huazai665]

I also encountered this problem. I found the liegroups do not support cuda.