PyTorch 설치

Pytorch

• Fackbook이 개발한 오픈소스 머신 러닝 라이브러리
• Python에 친화적인 라이브러리
• 최적화 모듈, 이미지 처리 모듈, 오디오 처리 모듈 등 다양한 모듈을 제공
• 클라우드 플랫폼 (Amazon Web Services, Google Cloud Platform)에도 손쉽게 적용 가능

 

ANACONDA

• 아나콘다는 운영체제 내 가상 Python 환경을 설정함
• 가상 환경 별로 간섭 없이 개별적으로 패지키를 유지 관리 할 수 있음

 

Pytorch CPU 설치

• 아나콘다를 통해 PyTorch를 설치할 수 있음
• 아나콘다를 사용하지 않는다면 PIP(Package Manager)를 통해 설치 가능
• Pythorch는 최적화 모듈, 이미지 처리 모듈, 오디오 처리 모듈 등 다양한 라이브러리를 제공함
  • pytorch : 심층 신경망 라이브러리
  • torchaudio : 오디오 및 신호 처리를 위한 라이브러리
  • torchvision : 컴퓨터 비전을 위한 이미지 변환 라이브러리
• Pytorch CPU(cpuonly)는 다음 명령어로 설치 가능

Conda Install

conda install pytorch torchvision torchaudio cpuonly -c pytorch

PIP Install

pip install pytorch cpuonly

 

PyTorch GPU 설치

• PyTorch GPU는 CUDA를 설치해야 활용 가능함
• 현재 PyTorch를 설치하려는 환경의 GPU 사양에 맞는 CUDA를 설치해야 함
• 그리고 해당 CUDA에 맞는 PyTorch를 설치해야 함
• GPU 사양 확인 (https://developer.nvidia.com/cuda-gpus#compute)

 

PyTorch 정상 설치 확인 방법

• 설치된 파이토치 패키지를 임포트하여 버전 확인

파이토치 버전 확인 PyTorch Version Check