pytorch, torchvision, cupy 설치 관련 사이트 및 설치 방법

Anaconda3 2020.11 버전을 설치하고 pytorch, torchvision, cupy를 설치하려 했는데 python 버전 문제 때문인지 설치가 되지 않았다.

 

그래서 conda 가상 환경을 python 3.7 버전으로 만든 뒤 설치를 해봤는데 제대로 설치가 되었다.

 

 

[conda 가상 환경을 만드는 방법]

1. 찾기에서 Anaconda Prompt (anaconda3)를 마우스 오른쪽 버튼을 눌러 관리자 권한으로 실행한다.

2. conda 명령문으로 가상환경을 생성한다.

   conda create --name 가상환경이름 python=x.x(버전) 

3. 패키지 설치를 할꺼냐고 물어보는데 y를 입력해준다.

4. 설치가 다 됐으면 생성한 가상환경을 활성화한다.

   activate 가상환경이름

5. (base)가 (가상환경이름)으로 바뀌었으면 가상환경이 활성화된 것이다.

 

※ 필요한 conda 명령문

- conda install 패키지이름

- conda uninstall 패키지이름

- deactivate 가상환경이름

- conda remove --name 가상환경이름 --all

 

 

(1) pytorch, torchvision 설치

pytorch, torchvision은 pytorch.org/get-started/locally/ 사이트에서 내 컴퓨터에 맞는 버전을 찾아 설치하였다. cuda 버전을 확인하려면 Window PowerShell이나 Anaconda Prompt 창에 nvcc --version를 입력하면 된다. (cuda를 설치하지 않았다면 developer.nvidia.com/cuda-toolkit에서 다운 받아야 한다.)

 

 

(2) cupy 설치

cupy는 numpy 형태의 데이터를 GPU로 계산하는 패키지다. CNN을 패키지를 이용하지 않고 구현해보기 위해 cupy가 필요했다. docs.cupy.dev/en/latest/install.html 사이트에서 설치하려는 컴퓨터에 있는 cuda 버전에 맞게 설치해야한다.

 

 

(3) 패키지 설치 확인

패키지가 제대로 설치 되었는지 확인하려면 python을 입력하여 python 프로그램에 들어간 뒤 import 패키지를 실행해보면 된다. python을 나오고 싶다면 exit()을 입력해야한다.