Apple Silicon(M1) 에 Anaconda & PyTorch 설치하기 + 성능 측정


• 애플 실리콘(M1)용 아나콘다 및 파이토치가 출시되어, 아나콘다 공식 홈페이지에서 M1용 설치 파일을 다운로드하여 설치할 수 있다.
• 파이토치를 설치한 후, M1에서 GPU 가속을 사용하려면 'cuda' 대신 'mps'에 텐서를 할당해야 한다.
• M1 GPU에서 PyTorch를 사용하여 MNIST 데이터셋을 분류하는 모델을 작성하고, Nvidia GPU, M1 GPU, 서버 CPU, M1 CPU 간의 속도를 비교한다.
• 속도 비교 결과, 예상대로 Nvidia GPU가 가장 빠르지만, M1 GPU의 성능도 상당히 높아 애플 실리콘 기반의 딥러닝 연산에 대한 가능성을 확인할 수 있다.

북마크
공유하기
신고하기