[Python] 파이토치(Pytorch) 설치하기

Pytorch란?

• 파이토치는 2017년 초에 공개된 딥러닝 프레임워크로 GPU를 활용하여 인공 신경망 모델을 만들고 학습시킬 수 있게 도와줌
• Numpy라이브러리에 비해 계산이 간단하고 병렬 연산에서 GPU를 사용하므로 CPU보다 속도면에서 월등히 빠름
• 유명한 딥러닝 프레임워크인 텐서플로(tensorflow)는 'Define and Run' 방식으로 연산 그래프를 먼저 만들고 실제 연산할 때 값을 전달하여 결과를 얻고 파이토치(Pytorch)는 'Define by Run' 방식으로 연산 그래프를 정의하는 것과 동시에 값도 초기화되어 연산이 이루어지고 연산 속도도 빠르고, 적어도 밀리지는 않음

 

Pytorch 설치하기

https://pytorch.org/

PyTorch

위의 공식 홈페이지에 접속합니다.

 

아래로 스크롤을 내리면 아래와 같은 INSTALL PYTORCH가 보입니다.

여기서 본인의 환경에 맞는 옵션을 선택합니다.

 

PyTorch Build는 파이토치의 어떤 버전을 사용할 것인지를 나타내고 Stable은 안정화된 버전, Preview는 현재 기준으로 가장 최신이지만 버그 등이 있을 수 있는 버전, LTS는 앞으로 버전업을 자주 하지 않을 경우 이 버전이 좋습니다.

 

저는 OS(Operation System)는 Mac을 사용하고 있고 Package는 아나콘다를 사용 중이기에 Conda를 선택했습니다.

 

2022.01.22 - [데이터분석 공부/Python] - [Python] Jupyter Notebook 설치 및 실행

[Python] Jupyter Notebook 설치 및 실행

파이썬의 아나콘다를 사용하기 위한 설치방법은 위에 첨부를 했습니다.

 

이어서 Language는 Python을 선택했고, Compute Platform은 파이토치의 장점인 GPU를 사용해서 CPU보다 속도를 빠르게 해주는 CUDA라는 API가 있지만 제가 사용하는 2020Macbook air는 지원이 되지 않아 Default인 CPU를 사용했습니다.

 

이제 선택에 맡게 뜨는 Run this Command의 conda install pytorch torchvision torchaudio -c pytorch를 맥의 터미널에서 실행하면 아래와 같이 실행되면서 완료됐다는 문구가 뜹니다.

 

이후 파이썬의 Jupyter Notebook에서 아래와 같이 import torch를 통해 파이토치를 불러오고 기능을 사용해보면

이렇게 오류 없이 실행되는 것을 볼 수 있습니다!