It seems Pytorch doesn't use GPU

I assume this points towards the shown GPU activity in the task manager.
If so, note that the Windows task manager doesn’t show the compute tab by default and you would have to select it via the drop-down menu.
I would recommend to stick to nvidia-smi to see the device utilization.

1 Like