Windows에 PyTorch 설치

대부분 아나콘다를 통해 설치를 권장하는데, 이미 파이썬 3.7이 설치되어 있는지라, 혹시나하여 아나콘다가 아닌 그냥 설치하는 방식으로 진행했습니다.

Python 3.7 설치

Python 사이트에서 해당 버전 설치 파일로 설치.

CUDA 설치

https://developer.nvidia.com/cuda-toolkit에서 최신버전(10.1)을 설치하였음.

cuDNN 설치

구글에서 cuDNN으로 검색하여 developer.nvidia.com에서 로그인하여 압축 파일을 다운로드 받고 CUDA가 설치된 경로에 복사.

파란색 부분이 cuDNN에 해당됨.

PyTorch 라이브러리 설치

https://pytorch.org/get-started/locally를 참고하여 pip3로 설치함.

위의 이미지를 보면 pip3 명령어를 이용해 cmd 창에서 설치가 이루어지는데, pip3는 별도 설치를 요할 수 있습니다.

답글 남기기

이메일 주소는 공개되지 않습니다. 필수 필드는 *로 표시됩니다