If model.to() was called with cuda:0 as its argument, the GPU will be used since PyTorch won’t move the parameters back to the CPU behind your back. Use nvidia-smi to check the GPU usage or the “CUDA” tab in the task manager, as it doesn’t show the compute utilization by default.