[ PyTorch ] PyTorch 설치하기( with 아나콘다)

Pytorch 설치 방법

1. PyTorch 홈페이지에서 pytorch 설치 commend 확인

pytorch.org/get-started/locally/

PyTorch

An open source deep learning platform that provides a seamless path from research prototyping to production deployment.

pytorch.org

Get Started를 클릭하면 화면에 START LOCALLY창이 보이고, 설정에 맞게 클릭 후 제일 하단에 있는 RUN this Command 를 확인한다.

※ CUDA 버전 확인 방법

>>>nvcc -V

>>>nvidia-smi

홈페이지에서 얻은 명령어를 가지고 pytorch를 설치한다. 2. 설치 진행

예시)

>>> conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

다음 명령어로 정상적으로 설치 완료된 것을 확인해 볼 수 있다.

>>> import pytorch

Green Chem