[Mac] pytorch gpu 설치하기

이제 Mac m1칩으로 pytorch gpu를 사용할 수 있다! (pyTorch 1.12부터 정식으로 지원)

 

- xcode와 miniforge3 설치 후에 진행

2023.01.14 - [setting/mac] - [Mac] Xcode, Miniforge3 설치

[Mac] Xcode, Miniforge3 설치

1. pytorch를 설치할 가상환경을 만들기

- miniforge3은 python 3.7 이하의 버전과 호환이 되지 않는다.

$ conda create -n pytorch python=3.8
$ conda activate pytorch

# 가상환경 목록 확인 
$ conda env list

2. pytorch 페이지에 접속해서 command 명령어를 복사

- Pytorch page : https://pytorch.org/

3. '1번'에서 만들어둔 가상환경에서 '2번'에서 복사한 명령어를 붙여넣어준다.

(pytorch) → ~/ conda install pytorch torchvision torchaudio -c pytorch-nightly

4. 터미널에서 바로 확인하는 방법

4-1. python 접속

(pytorch) → ~/ python

4-2. 두 개의 명령어를 통해 확인할 수 있다. 

- torch.backends.mps.is_available() : 현재 설치된 pytorch가 mps backend를 지원하는지 확인하는 방법  

- torch.backends.mps.is_built() : 사용할 수 있는 mps 장치가 있는지 확인 하는 방법

5. cuda ? mps ?

NVIDIA GPU 에서 cuda 장치를 이용한다면, Apple Silicon에서는 mps 장치를 이용한다.

6. 사용하는 방법 

- vscode, jupyter에서 "mps"장치를 지정한 뒤에 모델에 적용해서 사용하면 된다.

device = torch.device('mps') if torch.backends.mps.is_available() else torch.device('cpu')
print(device) # output : device(type='mps')