M1, M2 맥북에서 pytorch GPU 가속하기

Apple Silicon 칩에서 pytorch GPU 가속이 가능하게 되었습니다.

 

pytorch 1.12 버전부터 지원되며, 프리뷰 버전이 22년 6월 즈음에 나온 것 같습니다.

 

저는 M2 맥북에어로 진행했습니다.

 

1. miniconda3을 다운로드 받습니다. 아래 그림에서 “Miniconda3 macOS Apple M1 64-bit pkg”를 선택하시면 됩니다.

터미널을 띄우고 conda를 쳐보시면 설치가 되었음을 확인할 수 있습니다.

 

 

2. 필요한 여러 패키지 설치 (저는 python 버전을 3.10으로 정하였습니다.)

conda create -n torch python==3.10
conda activate torch
conda update --all
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 -c pytorch

 

 

3. GPU 가속 가능여부 확인

 

torch.backends.mps.is_available()

torch.backends.mps.is_built()

둘 다 True 가 출력되면 가속이 가능합니다!

 

CUDA 에서는 model.to(’cuda’) 이런식으로 코드를 짰었다면,

Apple silicon 에서는 model.to(’mps’) 이렇게 변경하면 GPU 가속이 가능합니다.

 

 

CIFAR-10을 이용해서 GPU 학습을 한 번 해보고 있는데, semaphore 누수 에러? 발생 때문에 삽질 중입니다.....

 

 

도움이 되셨다면 아무 광고나 클릭 한 번 부탁드립니다👍