[딥러닝] 파이토치 개발환경 구축

www.anaconda.com/products/individual 접속한다.

 

다운로드 클릭 후 설치파일을 관리자 권한으로 열어서 설치한다.

 

 

설치가 끝나면 프롬프트를 실행한다.

 

$ conda create -n pytorch python=3.7

 

입력하면 파이썬 3.7 기반의 pytorch라는 가상환경이 생긴다.

 

$ conda activate pytorch

 

가상환경을 켠다. 

 

developer.nvidia.com/cuda-toolkit-archive 에서 CUDA를 설치한다. 11.0으로 했다.

 

(왠지 모르겠는데 다운로드 버튼 클릭하면 안 먹하고 새 탭에서 열기 해야 다운로드 창이 열림. 왜지?)

 

그 와중에 마우스 프로세스 꺼야 한다고 해서 끄고 다른 마우스 끼웠다... 오래 걸린다.

 

pytorch.org/ 에서 설치한 CUDA 버전에 맞춰 파이토치를 설치한다.

 

아나콘다 프롬포트에 Run this Command 부분의 커맨드를 입력하면 된다. 파이썬 3.7이니까 윗 줄 무시하고 입력하면 된다.

 

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

 

오래 걸린다.

 

그동안 VS code에 파이썬 익스텐션도 설치해준다.

 

그러면 좌측 하단에 Python어쩌구.. 가 생기는데 이걸 누른 후 pytorch로 인터프리터를 변경해준다.

 

신기하다.