No, PyTorch only supports CUDA enabled devices(Nvidia GPUs) as GPUs.
You can still run PyTorch on your CPU.



 prateekazam:

Expected one of cpu, cuda, mkldnn, opengl, opencl, ideep, hip, msnpu, xla device type


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