
[PyTorch] M1, M2 칩으로 Pytorch GPU 사용하기
·
AI
글쓴이는 현재 M2 맥북 에어 환경에서 딥러닝 학습을 진행하고 있다. 본래 tensorflow 기반의 학습만 진행하다가 이번에 새로 pytorch 환경을 사용하게 되어서 Apple Silicon OS에서 pytorch 환경을 구축하게 되었다.cuda 환경은 MacOS와 호환이 불가하기 때문에, M1 M2 환경에서 pytorch를 사용할 수 있는 mps 장치 적용 방법에 대해서 알아보도록 하겠다. 💻 MPS란?MPS 백엔드는 MacOS 장치에서 PyTorch 에코시스템을 확장하고 GPU에서 작업을 설정하고 실행할 수 있는 기존 스크립트 기능을 제공한다. 1. torch version 확인하기터미널을 키고, python과 torch 라이브러리 설치가 우선적으로 진행되어야 한다.torch 1.12.0 이상..