반응형
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 버전 확인 및 호환이 잘되는 파이토치 버전을 찾는 방법은
아래의 포스팅을 참고해주세요.
환경마다 다르지만 커맨드 입력 후 설치에는 수 분 가량이 소요될 수 있습니다.
설치 후, 파이토치 import 및 GPU 확인이 성공적으로 된다면 설치가 완료된 것입니다.
(CPU 버전으로 설치한 경우에는 GPU 확인 결과가 False로 나옵니다.)
import torch
torch.cuda.is_available() # GPU 버전인 경우 True가 나오면 성공
'Python > Pytorch' 카테고리의 다른 글
[Pytorch] torch.view, torch.reshape의 사용법과 차이 비교 (1) | 2022.02.27 |
---|---|
[Pytorch] 파이토치 텐서 자료형 확인, 타입 변환(casting) 방법 (1) | 2022.02.08 |
[Pytorch] contiguous 원리와 의미 (1) | 2022.02.08 |