[linux] 그래픽 드라이버 & CUDA 설치

그래픽 카드 확인
lspci | grep -i VGA

사실 노트북 같은 경우는 저 명령어로 내장 그래픽이 나오는 경우도 있다.

NVIDIA그래픽카드가 뭔지 알고 있으면 위는 무시해도 된다.

설정에서 NVIDIA그래픽 카드 설정을 하는 경우도 있는데, 웬만하면 OS맛탱이 가니 아래의 방법으로 하자.


그래픽 카드 드라이버 확인

https://www.geforce.com/drivers

위 사이트에서 아래와 같이 검색한뒤 빨간 박스의 숫자를 기억한다.

나의 경우 384 가 드라이버 번호이다.

그래픽 카드 드라이버 설치
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-384 -y
sudo reboot
그래픽 카드 드라이버 설치 확인
nvidia-smi

CUDA 설치
sudo apt-get install linux-headers-$(uname -r)

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

위 사이트에서 자신의 환경에 맞는 파일 다운로드!

위에 나온 스크립트를 그대로 실행한다.

.bashrc 파일 에 경로 설정.

아래 내용을 ~/.bashrc 파일 맨 밑에 추가한다.

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
CUDA 설치 확인
nvcc -V
cuDNN 설치

https://developer.nvidia.com/cudnn
위에서 CUDA 와 같은 버전의 라이브러리를 다운로드 하고, 아래의 명령어 실행

만일 tensorflow 를 위해 cudnn을 설치할 계획이라면 6.0 이 아닌 5.1 버전을 받아야 한다.

tar xvzf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
sudo apt-get install libcupti-dev -y