파이썬 pytorch 설치하기

안녕하세요 오늘은 파이 토치(PyTorch) 설치 과정에 대하여 포스팅 하겠습니다.

저는 TextOCR 글자 인식 프로젝트 진행 중 파이 토치를 사용할 일이 있어 이번 기회를 통해 포스팅하게되었습니다.

1. PyTorch 홈페이지 https://pytorch.org/get-started/locally/

공식 PyTorch 홈페이지에서 Get Started를 정독하고 그에 따른 순서에 맞게 실행합니다.

2. 패키지 매니저 Anaconda 설치

저는 Anaconda를 이용하여 파이썬과 파이 토치 설치 하였습니다.

Anaconda download : https://www.anaconda.com/products/individual#windows

링크를 통해 홈페이지 접속 후 본인 컴퓨터에 맞는 비트 사양으로 Anaconda 다운.

아래 그림의 Download 클릭 후 64-Bit Graphical Installer 다운 설치합니다.

3. 본인의 python 버전, 컴퓨터 CUDA 버전, 지원 GPU 확인

python --verion : python 버전

nvcc --version : cuda 버전

cuda 버전이 없는 경우 아래 링크를 통해 지원 GPU확인 하고 설치합니다.

but, 지원하지 않아도 Pytorch 설치가능합니다.

CUDA 지원 GPU 확인 : https://developer.nvidia.com/cuda-gpus

4. CUDA 사용 유무 결정 후 CUDA Toolkit 설치 

CUDA Toolkit 버전은 아래 url을 통해 확인 가능 합니다.

최근 보다는 그 아래 버전이 좋을 것 같다는 판단에 따라 저는 10.0 버전을 설치하였습니다.

https://developer.nvidia.com/CUDA-TOOLKIT-ARCHIVE

5. Anaconda Prompt 창을 열고 PyTorch 설치

프롬프트를 열고 PyTorch 공식 사이트 맨 위에 선택 창에 따라 버전을 선택하고 torch를 설치.

Anaconda 프롬프트에 위의 Run this Command 라인을 쓰고 실행.

오류가 없이 실행이 잘 되었다면 상관없으니 오류 시 아래 밑줄 친 torchaudio 를 제거 후 cudatoolkit 버전을 다시 확인 하고 설치합니다.

변경 전

변경 후

6. 설치 후 import torch로 확인

위와 같이 import torch 실행 시 오류가 나지 않았다면 정상적으로 설치가 된 것임을 확인 할 수 있습니다.