PyTorch 1.12 with device "mps" for apple m1

[jeanjerome]

Hi,
PyTorch 1.12 is finally optimized for apple silicon M1 processor but it requires torch.device("mps") configuration.

I was not able to set it and got this error when trying to set device to "mps" :

RuntimeError: Expected one of cpu, cuda, xpu, mkldnn, opengl, opencl, ideep, hip, ve, ort, mlc, xla, lazy, vulkan, meta, hpu device type at start of device string: mps

Could you please provide such parameter settings?
Thanks.

[jeanjerome]

My bad, this is a pytorch-lightning issue.