[MacOS] M1에서 Pytorch 설치하기

Pytorch Stable version 설치하기

아래의 사이트에서 Pytorch를 설치할 수 있다.

https://pytorch.org/get-started/locally/

install code

conda install pytorch torchvision torchaudio -c pytorch

PyTorch - GPU Acceleration

device에 적용하기

import torch
device = torch.device('mps:0' if torch.backends.mps.is_available() else 'cpu')

MPS 지원 확인

print (f"PyTorch version:{torch.__version__}")
print(f"MPS 장치를 지원하도록 build 되었는지: {torch.backends.mps.is_built()}") 
print(f"MPS 장치가 사용 가능한지: {torch.backends.mps.is_available()}") 
!python -c 'import platform;print(platform.platform())'

위의 코드를 실행했을 때, Pytorch의 버전은 1.12.0 버전 이상이어야 하고, MPS is_built()와 is_available()이 모두 True로 나와야 한다.

예제 적용하기

import torch

device = torch.device('mps')
print(f" - device : {device}")
sample = torch.Tensor([[1, 2, 3], [4, 5, 6]])
sample = sample.to(device)
print(f" - gpu tensor : ")
print(sample)

 - device : mps
 - gpu tensor : 
tensor([[1., 2., 3.],
        [4., 5., 6.]], device='mps:0')