[Pytorch] Apple Silicon mac OS에서 Pytorch GPU 사용하기

파이토치(pytorch)를 주로 사용할 때는 CPU자원을 사용하거나 GPU 장비를 사용합니다. 하지만, Apple Silicon 맥북은 Nvidia-GPU 와는 다른 장비구조를 가지고 있습니다. 그래서 nvidia-gpu 사용하는 것과 동일하게 사용하면 에러가 발생하기 때문에, 파이토치를 apple silicon 맥북에서 사용하는 방법을 알아보려고 합니다.

 

Conda 설치

우선 개발 환경을 셋팅을 위해 conda를 설치해야 합니다.

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-arm64.sh
bash Miniconda3-latest-MacOSX-arm64.sh

 

Pytorch conda 버전 설치

그 후, conda 환경 + apple silicon 환경에 적합한 파이토치 라이브러리를 설치합니다.

 

 

사용 예제

코드 상에서 apple silicone 장비를 사용하기 위한 예제 코드는 아래와 같습니다.

import torch
device = torch.device("mps")