Hi @orionflame,
When you installed PyTorch did you install PyTorch with CUDA? It seems you may have installed the cpu only version of PyTorch, as torch.cuda.is_available()
returns False
.
Make sure to select an appropriate CUDA version for the PyTorch install via the getting-started page, here: Start Locally | PyTorch
These other threads on the forum might help you debug any issues: