Open arkinrc opened 2 years ago
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
I also encountered this problem. I found the liegroups do not support cuda.
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