Using Macbook GPU's with Pytorch

1

Hi All,

I have a new macbook and i was trying to setup pytorch on it. since this laptop doesn’t have NVIDIA gpu i was trying to work with MPS framework. but since i am completely new to this MPS thing how do i go about it ?

I have to use pytorch geometric. on first random try i was able to install everything and device was detecting MPS instead of cuda which meant my torch was able to use mac’s GPU, but when i ran some codes in torch_geometric i got some errors, something like

“operation is not supported on metal acceleration”,

can someone please show me the way for this configuration.

Thanks

2

There is still an open issue:

I assume that only some functionality is available.

I’ve been looking at their framework here, but am unsure yet whether its worth trying: GitHub - ml-explore/mlx: MLX: An array framework for Apple silicon

The syntax seems like PyTorch.

3

PyTorch on MPS works fabulously in my experience (training neural networks), but I don’t know about “geometric.” I’d would focus on geometric-specific debugging in your case. Maybe first confirm that you have more common PyTorch libraries working on MPS?