Ubuntu 20.04 NVIDIA GPU 드라이버 및 CUDA 설치

-. 준비
내장 포함 그래픽 정보
lspci -vnn | grep VGA

GPU 정보
lspci -vnn | grep NVIDIA

그래픽, GPU 세부 정보 확인
sudo lshw -numeric -C display

Ubuntu 기본 그래픽 비활성
sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"

내용을 확인한다.
cat /etc/modprobe.d/blacklist-nvidia-nouveau.conf

아래처럼 나온다.
blacklist nouveau options nouveau modeset=0

내용을 반영한다.
sudo update-initramfs -u

출력되는 W: Possible missing firmware /lib/firmware/... 같은 메시지지는 내장 그래픽에 관한 것이니 무시하고 기다린 후 프롬프트가 떨어지면 재부팅한다.

sudo reboot

-. 드라이버 설치
빌드할 패키지를 먼저 설치한다.
sudo apt install -y build-essential libglvnd-dev pkg-config
설치 파일을 다운로드한다. *설치에 사용한 파일의 최신 버전은 아래와 다를 수 있다.
wget https://us.download.nvidia.com/XFree86/Linux-x86_64/510.73.05/NVIDIA-Linux-x86_64-510.73.05.run

설치
sudo bash NVIDIA-Linux-x86_64-510.73.05.run
X 모듈 경로에 관한 경고는 무시한다.

 

재부팅
sudo reboot

GPU 정보를 확인한다.
nvidia-smi

-. CUDA 설치
설치 파일을 다운로드한다. *설치에 사용한 파일의 최신 버전은 아래와 다를 수 있다.
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
run 파일을 실행하여 설치를 시작한다.
sudo sh cuda_11.7.0_515.43.04_linux.run
창에서 accept를 입력하여 라이선스 사용에 동의한다.

 

창에서 Driver만 체크를 해제하고 다른 패키지들은 놔둔다. Install로 옮겨가서 엔터를 친다.

사용자 환경변수에 CUDA 경로를 설정한다.
vi ./.profile

if [ -d "/usr/local/cuda-11.7/bin/" ]; then
export PATH=/usr/local/cuda-11.7/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

적용한다.
source ./.profile 또는 sudo reboot

CUDA 정보 확인
nvcc --version