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

spring·2020년 11월 9일
0
post-custom-banner

######그래픽 카드 확인

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

profile
Researcher & Developer @ NAVER Corp | Designer @ HONGIK Univ.
post-custom-banner

0개의 댓글