Nvidia-driver, CUDA, CUDNN 설치

Auto-install로 쉽게 설치하는 방법

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo ubuntu-drivers autoinstall
sudo reboot

Nvidia-driver 수동 설치(Nvidia GPU notebook 이용자 추천)

우분투 18.04 - NVIDIA 드라이버를 설치하는 방법

Cuda 설치

우선 현재 그래픽 카드 및 드라이버에 맞는 Cuda 다운로드
- https://developer.nvidia.com/cuda-toolkit-archive
아래 명령어로 실행
- 중요한것은 cuda 설치 시 그래픽 드라이버 다시 설치할 것인지 묻는데 “No” 하기
- Run sudo sh cuda_XXX_XXX_linux.run
- Follow the command-line prompts
- accept -> no -> yes -> enter (example install)

CUDA 경로 설정

sudo gedit ~/.bashrc # open file

# 본인의 쿠다 버전에 맞게 명령어 설정
export PATH=/usr/local/**cuda-11.X**/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/**cuda-11.x**/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source ~/.bashrc # apply after save file

nvcc -V # cuda 설치되었는지 확인

Cudnn 설치

본인이 설치한 nvidia-driver / cuda 버전에 맞게 cudnn 버전 설치 (웬만하면 for CUDA 11.x 로 받아)
- https://developer.nvidia.com/rdp/cudnn-archive
- Local installer for Linux x86_64[Tar] 설치

설치한 tar 파일 디렉토리에서 압축 해제

$ tar -xvzf cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive.tar.xz

해제된 파일들을 local로 복사

$ sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.x/include 
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.x/lib64 
$ sudo chmod a+r /usr/local/cuda-11.x/include/cudnn*.h /usr/local/cuda-11.x/lib64/libcudnn*

cudnn 설치 확인

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

Untitled

심볼릭 링크 설정 (본인의 쿠다 버전에 맞게 아래 명령어 변경 )

$ sudo ln -sf /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_adv_train.so.8.x.x /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_adv_train.so.8

$ sudo ln -sf /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_ops_infer.so.8.x.x /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_ops_infer.so.8
 
$ sudo ln -sf /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_cnn_train.so.8.x.x /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_cnn_train.so.8
 
$ sudo ln -sf /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_adv_infer.so.8.x.x /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_adv_infer.so.8
 
$ sudo ln -sf /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_ops_train.so.8.x.x /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_ops_train.so.8

$ sudo ln -sf /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_cnn_infer.so.8.x.x /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcudnn_cnn_infer.so.8

심볼릭 링크 설정 확인

ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn

Untitled