but I do store tensors on GPU, when printing the tensors it says they’re on “device=cuda0”.
attaching another screenshot of task manager during the python run when GPU graph is enlarged.
the CPU usage went up to ~55% when started running but the GPU usage is at 0.

image.png2168×1860 232 KB