Python/Pytorch

[Pytorch] 파이토치 설치 방법 정리

jimmy_AI 2022. 2. 9. 20:05
반응형

torch install 방법 요약

안녕하세요. 이번 시간에는 매우 간단하게 사용 환경에 맞는 파이토치를

설치하는 방법을 말씀드리도록 하겠습니다.

 

 

1단계. 파이토치 사이트 접속 및 커맨드 입력

먼저, 파이토치 공식 사이트인 https://pytorch.org/에 접속해줍니다.

위와 같은 설정에서 각 항목을 아래 기준으로 선택해줍니다.

 

PyTorch Build : 현재 가장 안정 버전 / 최신 베타 버전 / 장기 지원 예전 버전 중 선택

Your OS : 리눅스 / 맥 / 윈도우 중 사용 운영체제 선택

Package : 가상환경이라면 Conda, 로컬환경이라면 Conda / Pip 중 선택 권장

Language : 사용할 언어 환경 선택

Compute Platform : GPU가 있다면 CUDA 버전에 맞게 선택, GPU가 없다면 CPU 선택

만일 CUDA 버전이 10.1이라면 =10.2 부분 커맨드를 =10.1 식으로 고쳐서 사용

 

이후 아래의 커맨드 메시지를 복사하여 터미널에 입력해주시거나

주피터 노트북에서 !를 앞에 붙여서 입력해주신 뒤 실행해주시면 됩니다.

 

 

2단계. 환경 체크 및 GPU 확인

참고로, CUDA 버전 확인 및 호환이 잘되는 파이토치 버전을 찾는 방법

아래의 포스팅을 참고해주세요.

 

[Pytorch] 쿠다 버전 확인, 파이토치 버전 체크, 업데이트 방법

CUDA 및 파이토치 버전 확인 및 관리 Pytorch를 사용하는 경우 CUDA 버전을 확인하고, 쿠다와 호환이 잘 되는 파이토치 버전으로 변경해주어야 하는 경우가 있을 수 있습니다. 이번 글에서는 간략하

jimmy-ai.tistory.com

 

환경마다 다르지만 커맨드 입력 후 설치에는 수 분 가량이 소요될 수 있습니다.

설치 후, 파이토치 import 및 GPU 확인이 성공적으로 된다면 설치가 완료된 것입니다.

(CPU 버전으로 설치한 경우에는 GPU 확인 결과가 False로 나옵니다.)

import torch

torch.cuda.is_available() # GPU 버전인 경우 True가 나오면 성공