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

 

Pytorch 설치하기 ( with. 아나콘다)

먼저 pytorch 홈페이지에 들어가서 Get Started 버튼을 누른다.

(여기로 바로 들어가도 된다.)

pytorch 홈페이지

 

그럼 위의 사진과 같은 화면이 뜨는데 이 때, CUDA 버전을 다음 명령어로 확인해야 한다. (이미 알고 있으면 안 해도 된다.)

> nvcc --version

cuda 버전 확인

또한, python 버전을 모르는 경우 python 버전도 확인한다.

> python --version

python 버전 확인

 

다시 위의 사진에서 맞는 python 버전과 CUDA 버전을 선택하면 아래의 Run this Command가 바뀌고, 이를 Anaconda prompt에 돌린다. (혹시 9.0이라면 9.2를 선택해서 cudatoolkit=9.2 부분을 9.0으로 바꾼다.)

        -> 이전의 CUDA 버전 PyTorch 설치

설치가 완료 되면, 다음 명령어를 쳤을 때 오류가 뜨지 않는다면 정상적으로 설치가 된 것이다.

$ python

>> import pytorch

 

 

Permission error

Anaconda Prompt를 관리자 권한으로 실행시켜 pytorch를 설치하면 해결 된다.

(리눅스의 경우 앞에 sudo를 붙인다.)