[Pytorch] Macbook M1에서 GPU사용하여 학습하기

로컬에서 학습을 돌릴 때, CPU로 하게되면 시간이 참 오래 걸리죠. 그런데, 맥북에는 GPU가 있는데, 이걸 쓰면 어떨까요??

Pytorch 버전 확인

파이토치는 최신 버전으로 설치해주시면됩니다

print (f"PyTorch version:{torch.__version__}")

버전을 확인하여 1.12.1 이상인지 확인해주세요.

GPU 사용 가능한 지 확인

만약 다음 코드를 실행하였을 때, 모두 True가 나온다면, GPU 사용 가능한 것 입니다.

print(f"MPS 장치를 지원하도록 build 되었는지: {torch.backends.mps.is_built()}")
print(f"MPS 장치가 사용 가능한지: {torch.backends.mps.is_available()}")

디바이스 추가

device = torch.device('mps:0' if torch.backends.mps.is_available() else 'cpu')

코드에 저렇게 한 줄 추가해주시면 됩니다.
사용할 때는 cuda 사용하듯이 해주시면 됩니다.

model.to(device)