맥북(Mac OS) 사용자의 Pytorch GPU를 사용법

#생성
conda create --name name_of_conda_env
#실행
conda activate name_of_conda_env
#만약 가상환경으로 돌아가고 싶다면
conda deactivate

import torch
#여기서 'cuda'가 아닌 'mps'를 사용합니다
device = torch.device('mps:0' if torch.backends.mps.is_available() else 'cpu')

print (f'PyTorch version:{torch.__version__}') # 1.12.1 이상
print(f'MPS 장치를 지원하도록 build 되었는지: {torch.backends.mps.is_built()}') # True 여야 합니다.
print(f'MPS 장치가 사용 가능한지: {torch.backends.mps.is_available()}') # True 여야 합니다.
!python -c 'import platform;print(platform.platform())'

model.to(device)

mps_device = torch.device('mps')
model.to(mps_device)