[Pytorch] pip로 Pytorch 설치하기

아래 파이토치 사용자 모임 사이트에서 자세할 설치방법을 볼 수 있습니다.

https://pytorch.kr/get-started/locally/#windows-python

파이토치 한국 사용자 모임 (PyTorch Korea User Group)

 

상황에 해당하는 설치 조건을 선택하면 그에 맞는 명령어를 표시해줍니다.

Windows - pip - Python - CUDA 11.6v 로 선택하였습니다.

※ CUDA? 프로그램 연산 시 GPU에 접근할 수 있게 해주는 플랫폼입니다. GPU가 NVIDIA 제품일 시 사용 가능합니다.

 

vscode를 기준으로 합니다.

터미널 입력란에 다음 명령어를 입력합니다.

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

 

다음과 같이 일련의 라이브러리 설치 후 pytorch 설치가 완료됩니다.

 

 

가상환경을 사용하는 경우 해당 가상환경 내에서 설치해야 합니다.

 

 

설치가 완료된 후 테스트 파일을 만들어 다음 코드를 입력한 뒤 실행합니다.

 

import torch
 
print(torch.rand(5, 3))
print(torch.cuda.is_available())

pytorch가 정상적으로 설치되었다면 다음과 같이 출력됩니다.

텐서의 값은 무작위이기 때문에 같지 않아도 됩니다.

line 4의 출력값이 true가 나온다면 해당 환경에서 GPU를 이용해 연산을 할 수 있다는 의미입니다.