PyTorch supported device


I have intel GPU, Can PyTorch support this? What other devices Pytorch support? Calling torch.device with the wrong value is giving below error.

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

Can I use the device from above?

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.

1 Like