Torch MPS on Intel Mac: get name of device?

619 Views Asked by At

In Python, on a machine that has an NVIDIA GPU, one can confirm the mps device name of the GPU using:

torch.cuda.get_device_name()

Per the docs: https://pytorch.org/docs/stable/generated/torch.cuda.get_device_name.html#torch.cuda.get_device_name

How do I confirm the mps device name for an Intel-based Mac, which has an AMD GPU? I don't see any mention of that in the docs.

0

There are 0 best solutions below