[꿀팁] Mac M1칩 맥북 GPU 사용하는법 code 코드

Mac M1칩 맥북 GPU 사용하는법 code 코드

엔비디아랑 애플이랑 사이가 안좋아서 못쓰게 어쩌구~~~~. 그런말이 있다.

그래서 cuda 말고 MPS를 이용해야 한다.

1) 먼저 맥북 gpu 사용 가능 여부 확인

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

 

2) 다음 코드로 실행

import torch

device = torch.device("mps" if torch.backends.mps.is_available() else "cpu")
if device.type == "mps":
    print("Using MPS (Metal Performance Shaders)")
elif torch.cuda.is_available():
    print("Using GPU")
else:
    print("Using CPU")