Use Apple GPU on PyTorch

Warning

Over a year old. May be outdated

Lesser Known Trick.

Use Apple GPU Acceleration (Metal Performance Shader) on PyTorch

if torch.cuda.is_available():
    print(f"Using NVIDIA GPU")
    self.device = torch.device("cuda")
elif torch.backends.mps.is_available():
    print(f"Using Metal Acceleration (MPS) on Apple GPU")
    self.device = torch.device("mps")
else:
    print("Using CPU")
    self.device = torch.device("cpu")

Backlinks1

240207