Hello, I am trying to use gpu for machine learning task from apple using "mps" as device for GPU but it is not working. I am using PyTorch Stable version. How can I use MacBook GPU for machine learning tasks?
Hi @mishrr - could you please share the error you are seeing and the Mac you are using?
I'm facing the same issue on MacBook Pro 14" with M3 Pro after creating Conda environment with Python 3.10 and installing PyTorch 2.2 for MacOs.
Both 'cpu' and 'mps' are using Performance Cores and not the GPU.
Any suggestions on how to use GPU or ANE for PyTorch?
