'python'에 해당되는 글 1건

  1. 2020.11.11 PyTorch 설치하기
2020. 11. 11. 10:55

PyTorch 설치하기

생각보다는 설명이 너무 간단하여 잘 찾지 못한 듯 하여, 다시 정리해 본다.

1. Ananconda 설치

Individual Edition 을 download 받아서 설치한다. (아래 이미지를 누르면 Anaconda install 사이트로 이동). 나는 Windows 64 bit 를 설치했다. 

 

2. PyTorch 설치

pytorch.org 에 가면, Start Locally 에서 설치 방법을 알 수 있다. (아래 사이트)

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

여기에서 원하는 조건을 선택하면, 아래에 명령이 나온다. 이 명령을 실행하면 설치 완료.

(폴더의 권한 때문에, "관리자 권한" 으로 먼저 실행할 필요가 있다. 나는 아래 조건으로 선택하여 설치함)

3. jupyter notebook 재설치

4. jupyter notebook 재 실행하여 확인. OK

이상입니다. 알고나니 쉽네요. ㅎㅎㅎ

 

<그런데, 사용해 보니, CUDA 를 사용하는 VGA는 TITAN 이상인 것 같군요 다 지우고 cpuonly로 재설치>

5. uninstall

(base) PS C:\WINDOWS\system32> conda uninstall pytorch torchvision cudatoolkit -c pytorch

6. pytorch 재설치 (w/ cpuonly) 및 jypyter notebook 재설치

(base) PS C:\WINDOWS\system32> conda install pytorch torchvision torchaudio cpuonly -c pytorch

(base) PS C:\WINDOWS\system32> conda install jupyter

이상입니다. 혼란이 있었네요 (CUDA를 사용할 수 없다는 것을 모르고)..ㅎㅎㅎ