PyTorch 설치

PyTorch를 설치하기 위해서는

   1.  PyTorch 홈페이지(pytorch.org)에서 사용자 환경을 선택한 후

   2.  생성된 명령어를 복사, 실행하면 된다.

 

아나콘다에서 가상환경을 먼저 설치한 후 PyTorch를 설치한다. 만약 GPU환경에서 사용할 것이라면 CUDA와 cuDNN을 먼저 설치해야 한다.

 

※ CUDA, cuDNN 설치와 가상환경 설치는 아래링크를 참고할 것

https://zmade.tistory.com/31?category=1203705 

아나콘다(파이썬 3.8) + 텐서플로 2.5 설치

 

PyTorch 홈페이지

http://pytorch.org 

PyTorch

 

PyTorch 설치

홈페이지에서 스크롤 다운하면 아래와 같은 화면이 보인다. 웹에서 홈페이지에 접속한 내 컴퓨터 사양을 확인하고 설치할 수 있는 항목을 기본으로 세팅해준다.

 

 

개인적으로 수정할 항목을 선택하면, "Run this Command"의 설치 명령어가 수정한 내용에 맞게 바뀌는데, 설치할 환경에서 해당 명령어를 수행하면 PyTorch가 설치된다.

위의 예는 콘다 가상환경에 설치하는 것으로 선택되었으므로, "Run this Command"의 명령어를 복사하여 콘다 가상환경에서  실행하면 설치된다.