You said that you had a successful start utilizing the GPU, and then it shifted to the CPU. Can you get it started running on the GPU again, or is it since that moment just stuck on the CPU instead?
When you did that CUDA reinstall, did you install drivers compatible with your Torch version? I made the mistake of installing the latest CUDA drivers before, and had to downgrade since PyTorch wasn’t up-to-date with those.