Pytorch Training Fails on Apple M2 - device = "mps"

Machine Learning & AI General
narenq7 OP
Created Jul ’23
Replies 1
Boosts 0
Views 713
Participants 2

I'm trying to train a simple GNN on Cora dataset which is really small. I use Jupyter notebook. I get dead kernel every time. When I switch to device ="cpu" it runs quickly. Am I missing something while using MPS

Replies  1
Boosts  0
Views  713
Participants  2
Graphics and Games Engineer OP
Apple
Jul ’23

Hi @narenq7 - could you please share the PyTorch version you are using (from pip list)? Also, if you could share a small sample reproducing the issue, it would be very useful.

0
Pytorch Training Fails on Apple M2 - device = "mps"
First post date Last post date
 
 
Q