맥북아, 친해지자 - (4) M1에서 Pytorch를 사용해보자

그토록 기다렸던 M1 Mac GPU 사용이 가능한 Pytorch 업데이트가 공개 되었다.

 

M1 Mac에서 GPU 사용하는 방법을 알아보자.

참고로 아직 정식 버전이 아닌 Preview 버전이므로, 가상환경을 통해 진행하는 것을 권장한다.

 

우선, 가장 많이 헤맸던 부분인데, 맥 버전을 최신형으로 업데이트 해주자.

현재 나의 맥 버전은 12.1

12.3 이상이여야 나중에 torch.backends.mps.is_available()에서 False가 아닌 True가 나올 수 있다.

 

그 후 Anaconda를 재설치해줘야 하는데, ARM 64bit가 지원되는 버전으로 다시 설치해줘야 한다.

우선 기존의 Anaconda를 아래 방법을 통해 지워주자.

conda install anaconda-clean

anaconda-clean --yes

rm -rf ~/anaconda3

# .bashrc 에서 conda와 관련된 것 전부 주석처리

 

그 후 anaconda 홈페이지에 들어가서 64-Bit (M1) Command Line Installer(420 MB)를 설치해준다.

설치가 완료되었으면 아래 코드를 통해 아나콘다를 재설치 해줄 수 있다.

cd Downloads
bash Anaconda3-2022.05-MacOSX-arm64.sh

아나콘다 재설치가 끝났으면 이젠 파이토치를 설치해줄 차례이다.

conda create -n gpu_1 --clone base # base에 있는 라이브러리를 gpu_1이라는 이름의 가상환경으로 복제
conda activate gpu_1 # 가상환경 실행
pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu

설치가 완료되었다면, 파이썬을 실행한 후 아래와 같이 확인해보자.

>>> import torch

>>> print(torch.__version__)

1.13.0.dev20220709

>>> print(torch.backends.mps.is_available())

True

>>> print(torch.backends.mps.is_built())

True

 

둘 다 True가 나왔다면 정상적으로 설치되었음을 알 수 있다.