파이토치(PyTorch) 설치

파이토치(PyTorch) 설치

---

파이토치를 install 할 때는 아래 페이지 (파이토치 공식 홈페이지) 에서 자신의 PC 환경과 동일하게 설정을 해주면 command를 생성해준다.

https://pytorch.org/

PyTorch

그런데 여기서 Compute Platform 항목에 CUDA 라는 것이 존재한다. CUDA란 Compute Unified Device Architecture의 약자로 NVIDIA가 만든 병렬 컴퓨팅 플랫폼 및 API 모델이다. CUDA 플랫폼은 GPU의 가상 명령어셋을 사용할 수 있도록 만들어주는 소프트웨어 레이어이며, NVIDIA가 만든 CUDA 코어가 장착된 GPU에서 작동한다. 즉, GPU를 사용하기 위해서는 CUDA가 필요한 것이며, GPU 사용 없이 파이토치를 사용한다면 Compute Platform 항목에서 CPU를 선택해주면 된다. 어떤 CUDA를 선택해야 하는지 확인하려면 우선 자신의 PC GPU 상태를 먼저 알아야 한다.

 

GPU 확인

---

1. 내 PC의 속성을 클릭 한다.

 

2. 장치 관리자를 선택한다.

 

3. 디스플레이 어댑터 안에 그래픽 카드 정보가 있다. 드라이버가 제대로 설치되어 있지 않으면 이 부분이 ? 로 표시된다.

이렇게 GPU가 확인 되었으면 GPU에 맞는 CUDA 버전을 찾아야 한다.

 

CUDA 버전 확인

---

구글에 본인의 GPU 이름 + cuda version 이라고 검색하면 왠만한 내용을 다 찾을 수 있을 것이다. 검색 와중에 누군가 정리해 놓은 사이트가 있어서 아래 공유한다. 나는 NVIDIA TITAN Xp 이므로 CUDA 8+, 8 이상은 모두 지원이 되는 것 같다.

https://gist.github.com/standaloneSA/99788f30466516dbcc00338b36ad5acf

Nvidia GPUs sorted by CUDA cores

 

CUDA 설치

---

다시 https://pytorch.org/ 로 돌아가서 내 PC에 맞는 버전을 선택하고 이에 따라 나타나는 command를 입력하여 파이토치를 설치한다. 성공적으로 설치 된 것을 확인할 수 있다.