반응형

MacBook M1에 PyTorch를 설치하는 과정을 다루고 있습니다. 



다음 링크를 참고하여 작성했습니다.

https://betterprogramming.pub/how-to-install-pytorch-on-apple-m1-series-512b3ad9bc6



2021. 5. 2.  최초작성



0. 다음 포스트를 참고하여 Xcode를 설치합니다. 

 

MacBook M1에 Xcode 설치하기

https://webnautes.tistory.com/1570




1. 다음 포스트대로 Macbook M1에 Miniforge를 설치하여 Python 개발환경을 만든 후,  Visual Studio Code와 연동하는 작업까지 진행합니다. 

 

Miniforge는 Anaconda, Miniconda처럼 Python 가상 환경을 만들어 주는 역할을 하는 것입니다.



Macbook M1에 Miniforge 설치하여 Visual Studio Code와 연동하기

https://webnautes.tistory.com/1638 



2. 주피터 노트북을 설치하여 Visual Studio Code와 연동합니다.

 

Macbook M1에 Visual Studio Code + Jupyter Notebook 개발 환경 만들기

https://webnautes.tistory.com/1664 



3. Command + 스페이스바를 누른 후, terminal.app를 입력하고 엔터를 눌러 터미널을 실행합니다. 

 

PyTorch를 설치하여 사용할 Python 가상환경을 생성합니다. 여기에선 Python 3.8를 사용하도록 했습니다. 

 

conda create -n pytorch_py3.8 python=3.8



4. 가상환경을 활성화 합니다.

 

conda activate pytorch_py3.8



5. PyTorch를 설치합니다.

 

아래 링크에서 다음처럼 선택하면 PyTorch 설치시 필요한 명령을 알려줍니다.

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

 

현재 Conda 패키지는 M1용이 없기때문에 Pip를 선택해야 합니다.  

글작성시점에서 PyTorch에서 MacBook M1의 GPU를 지원하지 않기때문에 Compute Platform이 CPU로 선택되어 있습니다. 

 



Conda를 선택해보면 다음과 같은 메시지를 볼 수 있습니다.

 

# MacOS Conda binaries are for x86_64 only, for M1 please use wheels

conda install pytorch torchvision torchaudio -c pytorch




위에서 확인한  명령을 사용하여 설치를 진행합니다.

 

pip3 install torch torchvision torchaudio



6. 터미널에서 다음 명령들을 입력하여  설치된 PyTorch를 확인해봅니다. 

 

(pytorch_py3.8) webnautes@webnautesui-MacBookAir ~ % python

Python 3.8.13 | packaged by conda-forge | (default, Mar 25 2022, 06:05:16) 

[Clang 12.0.1 ] on darwin

Type "help", "copyright", "credits" or "license" for more information.

>>> import torch

>>> torch.__version__

'1.11.0'



7. Visual Studio Code에서 간단한 PyTorch 코드를 실행해봅니다. 



Visual Studio Code에서 새로운 폴더를 열어줍니다. 

 



8. 주피터 노트북을 사용하여 테스트해봅니다. 확장자가 ipynb 소스 코드파일을 하나 생성합니다. 

 



9.오른쪽 위에 보이는 Select Kernel을 선택하여 

 



앞에서 생성한 가상환경을 선택해줍니다.

 



9. 다음 코드를 입력한 후, 빨간색 사각형의 아이콘을 클릭합니다.

 



다음 메시지박스가 보이면 Install을 클릭합니다. 

 



설치된 PyTorch 버전이 출력됩니다. 







반응형

해보고 확인한 것을 문서화하여 기록합니다.


부족함이 있지만 도움이 되었으면 합니다.
잘못된 부분이나 개선점을 알려주시면 감사하겠습니다.



포스트 작성시에는 문제 없었지만 이후 문제가 생길 수 있습니다.
문제가 생기면 포스트와 바뀐 환경이 있나 먼저 확인해보세요.

질문을 남겨주면 가능한 빨리 답변드립니다.


제가 쓴 책도 한번 검토해보세요 ^^

  • 네이버 블러그 공유하기
  • 네이버 밴드에 공유하기
  • 페이스북 공유하기
  • 카카오스토리 공유하기

댓글을 달아 주세요

">
  1. thumbnail
    Favicon of https://masked-the-hunter.tistory.com BlogIcon 가면 쓴 사냥꾼

    감사합니다. 덕분에 설치 잘 마쳤습니다.