[M1] Mac M1 pytorch GPU 사용하기

% Pytorch 1.12.0 이상의 버전에서 mps를 지원한다. 

import torch

device = torch.device(device='mps')
# model.to(device)
print(torch.backends.mps.is_available())

True 

 

M1 활성 상태 보기

Spolite 검색 -> 활성 상태 보기 -> GPU 기록(cmd+4)