Mac에서 torch GPU사용 : [x]cuda, [o]mps

Mac에서 pytorch 사용 시, GPU 가속을 하는 방법을 정리하였습니다.

 

Mac에서는 nvidia GPU를 사용하지 않기 때문에, cuda 사용이 불가능합니다.

따라서, Mac에서는 cuda가 아닌 mps를 사용하여 가속을 할 수 있습니다.

 

다행히, pytorch 홈페이지에서 설치에 대한 설명을 제공합니다.

PyTorch

 

위의 URL로 접속을 한 뒤, 아래와 같이 preview로 설치를 해줍니다.

참고로, Stable은 안정화된 버전이고, Preview(Nightly)는 nightly 빌드버전으로, 현재 기준 가장 최신의 버전입니다. 다만 버그 등이 있을 수 있어, Stable보다는 불안정한 버전이라고 생각해주시면 됩니다.

하지만, 우리는 mps 가속을 적용하기위해서 Preview(Nightly)로 설치해줍니다.

 

또한, pip install 시에는 로컬환경에 따라 자잘한 에러가 발생 할 수 있으므로,

되도록 conda install로 설치를 해주는게 좋습니다.

 

마지막으로, 설치 후에는 아래와 같이 cuda와 동일한 방식으로 mps를 적용시켜주면 됩니다.

# cuda
device = 'cuda' if torch.cuda.is_available else 'cpu'

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

# 가속설정
model.to(device)