Ubuntu 16.04 nvidia driver setup, CUDA setup

in #busy6 years ago

몇 번 설치를 해봤지만, 항상 짜증나는 일이다.

그래픽 카드 확인하기

lspci | grep VGA

01:00.0 VGA compatible controller: NVIDIA Corporation GK110 [GeForce GTX TITAN] (rev a1)

근데 요즘은 cuda-kit만 설치하면 자동으로 nvidia-driver도 설치된다.

https://developer.nvidia.com/cuda-downloads에서 하라는대로 한다.

간단해졌다. 아래와 같이 체크를 한다.

lspci -k 확인

kernel driver in use: nvidia

kernel_modules: nvidia_410(XXX)

nvidia-smi 확인

PATH, LD_LIBRARY_PATH에 cuda 경로 추가하기

export PATH=/usr/local/cuda-10.0/bin:${PATH:+${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:${LD_LIBRARY_PATH:+${LD_LIBRARY_PATH}}

nvcc -V 체크