본문 바로가기
개인 공부/연결, 설치

[PyTorch] 에어 M1 파이토치(PyTorch) 설치

by 육츠 2024. 9. 3.
Contents 접기

구직 사이트를 보면 파이토치가 보이던데 어떤 건지 궁금해서 일단 설치.

이 노트북 생각보다 설치 제약이 너무 많다,,

 

파이토치란

딥 러닝과 기계 학습을 위한 오픈 소스 프레임워크

- Python 기반의 과학 계산 패키지인 토치(Torch)를 기반으로 개발됨
- 텐서 연산과 자동 미분을 위한 기능을 제공하여 신경망 모델의 구성, 학습, 추론을 쉽게 구현할 수 있다.

 

파이토치 설치

참고 : bolero2.log

 

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

M1 macbook에 Anaconda native와 pytorch native 설치 및 속도 비교

velog.io

 

준비사항 : '아나콘다' 지만, 나는 이미 깔려있기 때문에 따로 적어놓지는 않았다.

사이트: https://pytorch.org/get-started/locally/

여기서는 Nightly로 설치해야한다고 한다. (공식문서에 쓰여있다고 적혀 있음)

pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu

 

PyTorch 설치 완료!

 

여담으로 ..
처음에는 Homebrew 를 사용한 설치 방법을 사용했다. 하지만, 아래 사진과 같이 x86_64 와 arm64 문제 때문에 다른 방법을 찾은 것