[PyTorch] PyTorch 설치

 

1. PyTorch 다운로드 사이트 접속 (https://pytorch.org/get-started/locally/)

 

2. 아래 사진과 같은 창에서 원하는 버전, 운영체제 등을 선택

 

3. *Conda Package 기준* conda prompt 실행 후 아래 명령어 입력

# cudatoolkit 부분은 빼도 설치가 가능함
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

 

4. Jupyter Notebook에서 import torch 명령어를 이용해 잘 로드되는지 확인하고 이상이 없을 경우 설치 끝!