pykan icon indicating copy to clipboard operation
pykan copied to clipboard

Published 20 hours ago • verified publisher icon KindXiaoming

Using CUDA couldn't work

Open GuoMinghui07 opened this issue 7 months ago • 6 comments

截屏2024-07-19 08 07 13

When I try to train the model on GPU: RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

There seems to be a problem with the internal function implementation.

GuoMinghui07 avatar Jul 19 '24 00:07 GuoMinghui07