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

alikeayal · 2024-10-21T12:20:13Z

Use MultKAN.py and raise this Error.

I check the debug, The variable x is on CUDA: 1, while variables such as subnode_stcale are on CUDA: 0.

I just updated pykan today, starting from version 0.0.5.
I was planning to try using KAN on multiple GPUs, but unexpectedly encountered this issue.
I hope someone can help me solve this problem. Thank you！ @KindXiaoming

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

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

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

alikeayal commented Oct 21, 2024 •

edited

Loading

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

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

Comments

alikeayal commented Oct 21, 2024 • edited Loading

alikeayal commented Oct 21, 2024 •

edited

Loading