PyTorch supported device

Hi,

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?

1 Like

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

What about Apple Silicon or AMD or Intel Arc? Can the pytorch leverage these GPUs?

I don’t believe the binaries are upstreamed in pytorch yet but you can find your way poking around here Stable Diffusion with Intel® Arc™ GPUs Using PyTorch and Docker

1 Like