아가 뭐해? - 컴퓨터 셋팅(2) NVIDIA 그래픽드라이버/cuda9.0/cudnn 설치하기

yolov3를 cpu로만 진행하면 실시간 웹캠으로 테스트를 진행할 때 7~8분에 한 번씩 프레임이 바뀔정도로 느려서

빠른 속도를 위해 gpu를 사용해야합니다.

그러기 위해서 엔비디아 그래픽드라이버, cuda, cudnn을 설치해줄건데,

설치는 NVIDIA 그래픽드라이버 -> CUDA -> CUDNN 순서로 진행됩니다.

모두 각자 컴퓨터에 호환되는 버전으로 다운받아줘야되는데 버전 확인법을 알려드리겠습니다.

맞지 않는 버전을 다운받게되면 삭제하고 다시 받아야되는 귀찮은 일이 발생하므로 미리 확실히 확인 후 다운 받는 것을 추천드립니다.

 

그래픽드라이버 버전 확인법

저는 GeForce 840M 이므로 네이버에 검색 후 들어가서 버전을 확인합니다.

 

384로 나와있죠 ?

 

https://www.nvidia.co.kr/Download/Find.aspx?lang=kr 

여기서도 확인 가능.

 

cuda 버전 확인 법

다운받은 그래픽드라이버 버전에 맞는 CUDA 버전이 나와있으니 보시고 하면됩니다.

https://docs.nvidia.com/deploy/cuda-compatibility/

CUDA Compatibility :: GPU Deployment and Management Documentation

 

저의 nvidia 그래픽드라이버는 384 버전이므로, cuda 9.0을 설치해주었습니다.

 

Cudnn 버전확인법

cudnn은 다운받을 사이트 들어가면 쿠다버전과 함께 나오니 그건 그때 확인하기 !

 

 

---

1. NVIDIA 그래픽 드라이버 다운받기

1-1. 먼저, Ctrl+Alt+T를 눌러 터미널을 열어 업그레이드 및 업데이트를 해줍니다.

sudo apt-get upgrade
sudo apt-get update

 

1-2. 소프트웨어&업데이트 로 들어갑니다.

'추가드라이버'에 들어가 각자 맞는 버전클릭 후 Apply Changes(바뀐내용적용) 클릭

 

이런식으로 떠야 정상입니다.

 

 

'No additional drivers available' 이 뜬다면 노트북에 NVIDIA 그래픽 드라이버가 장착되어있는지 확인해주세요.

만약 장착되어있는데도 이렇게 뜬다면 구글링 해보는 것을 추천

 

 

 

1-3. 재부팅 후 아래 코드를 터미널에 실행

nvidia-smi

 

이런식으로 뜬다면 설치가 잘 된 것.

 

 

---

2. CUDA 설치하기

CUDA Toolkit 설치하는 홈페이지입니다.

각자 호환되는 버전에 맞게 설치해주시면됩니다.

https://developer.nvidia.com/cuda-toolkit

CUDA Toolkit

 

우분투 버전만 각자에 맞게 선택해주면 됩니다.

 

선택 후 맨위에 뜨는 Base Installer의 Download 클릭

 

 

설치 후 터미널을 열어 다운한 폴더로 들어가 아래 코드 두줄을 입력해줍니다.

다운한 폴더로 들어가는 방법은 폴더명이 ujin 인 경우, 터미널 창에

cd ujin

을 해주면 됩니다. 

~$ chmod +x cuda_9.0.176_384.81_linux.run
~$ sudo ./cuda_9.0.176_384.81_linux.run

위의 코드를 입력하면 엄청난 긴 글들이 나오는데 읽을필요 없이 Ctrl+C 후 accept 해주면 됩니다.

후의 물음에는 이렇게 대답하면 됩니다.

n - y - y - y - enter 후 재부팅

 

 

 

재부팅 후 터미널을 켠 후 아래 코드 실행

vim .bashrc

안된다면 앞에 sudo를 입력 후 다시 실행

하면 아래와 같은 화면이 나옵니다.

 

 

 

맨 아랫부분에 두줄 추가 후 esc + :wq +엔터 (linux에서 저장하는 법)

export PATH=/usr/local/cuda/bin:$PATH  
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

 

다시 터미널로 나왔으면 아래코드 실행

source .bashrc

 

마지막으로 nvcc -V 혹은 nvcc --version 입력하면 설치가 된 것을 확인할 수 있습니다.

 

 

---

3. cuDNN 설치하기

 

cuDNN을 설치하기 위해서는 회원가입이 필요합니다.

https://developer.nvidia.com/rdp/cudnn-archive

cuDNN Archive

회원가입 후 각자 맞는 버전에 따라 설치해주세요.

for linux 로 설치해야합니다.

 

 

 

설치 후 아래와 같은 코드를 터미널에 입력합니다.

~$ cd [cuDNN을 다운로드 받은 경로]
~$ sudo tar -xzvf [다운받은 파일이름]
~$ cd cuda
~$ sudo cp include/cudnn.h /usr/local/cuda/include
~$ sudo cp lib64/libcudnn* /usr/local/cuda/lib64
~$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
~$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

 

마지막 줄을 입력했을 때 이렇게 나오면 설치 성공입니다.