pytorch 설치 방법

pytorch란?

pytorch는 기존에 딥러닝과 관련된 오픈소스 라이브러리이다. 기존에 torch가 있던 것을 python을 이용해 좀 더 쉽게 사용가능하도록 다시 만든 것을 pytorch로 명칭했다.

pytorch 사용 이유

python을 기반으로 하고 있어 코드를 익히거나 구현하는데 좀 더 쉽다는 평이 있다.
GPU를 이용한 tensor 계산으로 성능적으로도 우수한 측면이 있다.

pytorch 설치하기

설치방법도 매우 간단하다.
아래 사이트에 접속하면 UI로 사용자들이 어떤 환경에서 pytorch를 사용하는지 조합을 선택할 수 있고 그에 맞는 명령어를 제공해 준다.

 

https://pytorch.org/

머신러닝, 딥러닝 관련 개발자들은 conda를 많이 사용할텐데 파이썬 패키지 다운로드를 위해서 pip를 선택해서 진행해도 무관하다.
자신이 사용하는 컴퓨터 환경에 맞게 잘 선택해주자.

pytorch 설치 확인하기

pytorch 설치 확인을 하는 방법은 python을 열어서 실행해도 되고 jupyter를 열어서 실행해도 됩니다.

아래 torch를 import 한 후 기본 코드를 작성해본 뒤 실행되면 완료!

import torch