Solved by vaisakh_m in post #2 
              
              
                No, PyTorch only supports CUDA enabled devices(Nvidia GPUs) as GPUs. 
You can still run PyTorch on your CPU. 

All the devices mentioned here are compatible with PyTorch. 
Also, check if this is useful.