본문 바로가기
컴퓨터

파이토치(pytorch) 설치 방법

by skyjwoo 2020. 1. 31.
728x90
반응형

내가 자주 찾아보는 딥러닝 커뮤니티에서 pytorch 이야기가 많이 나오길래, 또, pytorch를 기반으로한 논문들도 많이 나오고 있기에, 이번 기회에 pytorch에 대해 한 번 공부해 보고자 하였다/

.

역시 가장 먼저 할 일은 pytorch를 설치하는 것인데, anaconda가 깔려있으면 매우 쉽게 설치할 수 있다. 

 

아나콘다가 설치되어 있지 않다면 아래 링크를 통해 설치할 수 있다. 

 

https://www.anaconda.com/distribution/

 

Anaconda Python/R Distribution - Free Download

Anaconda Distribution is the world's most popular Python data science platform. Download the free version to access over 1500 data science packages and manage libraries and dependencies with Conda.

www.anaconda.com

 

아나콘다가 설치되어있다면, 아래 링크에서(pytorch 홈페이지) pytorch 설치에 관한 정보를 얻을 수 있다. 

 

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

 

PyTorch

An open source deep learning platform that provides a seamless path from research prototyping to production deployment.

pytorch.org

 

설치 하고 싶은 환경을 선택해 주고, 아래에 나온 command를 실행시켜 주면 된다. 

필자는 windows os에서 anaconda package에 python에다 cuda를 사용하지 않음을 체크해 주었다. 

 

그리고 밑에 나온

conda install pytorch torchvision cpuonly -c pytorch

를 anaconda prompt에 실행시켜 주었다. 

설치 후 jupyter notebook에서 pytorch가 잘 불러와지는 것을 확인할 수 있다. 

728x90
반응형

댓글