[WINDOWS10 환경] 딥러닝을 위한 파이토치(pytorch) 설치 방법

설치하기

1. PyTorch 공식 사이트 들어가기 : https://pytorch.org/

2. Prvious versions of PyTorch1 클릭

3. CUDA 버전에 맞는 pytorch 설치

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

4. 실행창에 코드 넣고 enter

pytorch 실행시 GPU 작동 확인

import torch

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

print('Device:', device)
print('Current cuda device:', torch.cuda.current_device())
print('Count of using GPUs:', torch.cuda.device_count())