Migration Problem of Code on Apple M1 Chip

[Elody-Chen]

Hi, in the process of testing the code, I found that the function .cuda() of this line cannot run on M1 chip. Previously, PyTorch was able to use M1 chips for GPU acceleration, but this line did not work. May I ask if you have any solutions?

[kanswer]

.cuda() cannot be used in M1 chip, instead which u should use gpu like "x = torch.ones((1024 * 12, 1024 * 12), dtype=torch.float32,device='mps')".