Install pytorch with cuda. This is a complete guide to install PyTorch GPU on Windows.