While using MPS backend in PyTorch, I've found out that there is no way to select a GPU. This feature would be really beneficial while running Mac Pro with multiple GPU's.
We currently don’t have multi-GPU support.
We currently don’t have multi-GPU support.