Ubuntu 20.04 Cuda Setting

Ubuntu 기본 Setting

$ sudo apt update

$ sudo apt-get install build-essential

$ sudo apt-get install -y gnome-tweak-tool

$ sudo apt-get install vim

install Nvidia-driver

설치 가능 driver 확인

$ ubuntu-drivers devices

현재 사용 중인 그래픽카드 확인

$ lspci | grep -i nvidia

원하는 driver version 설치

$ sudo apt install nvidia-driver-515

$ sudo apt update

$ sudo apt upgrade

$ sudo reboot

nvidia-driver 설치 확인

$ sudo lshw -c display

driver=nvidia인지 확인

$ nvidia-smi

Install CUDA Toolkit

Driver와 맞는 Cuda version 확인

Release Note - CUDA Tookit Documentation

Cuda Tookit 설치

Cuda Tookit 11.7 Downloads

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

$ wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb

$ sudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb

$ sudo cp /var/cuda-repo-ubuntu2004-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/cuda-archive-keyring.gpg

$ sudo apt-get update

$ sudo apt-get -y install cuda

CUDA 환경변수 추가

$ vim ~/.bashrc

insert 키 누른 후 아래 코드 추가

# <<< conda initialize <<<
export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64

PATH=${CUDA_HOME}/bin:${PATH}
export PATH

Esc 키 누른 후 :wq로 저장 후 나가기

$ source ~/.bashrc

CUDA Tookit 설치 확인

$ nvcc -V

Install CuDNN

CuDNN Downloads

파일 복사

$ sudo cp ~/cudnn-linux-x86_64-8.5.0.96_cuda11-archive/include/* /usr/local/cuda-11.7/include

$ sudo cp ~/cudnn-linux-x86_64-8.5.0.96_cuda11-archive/lib/* /usr/local/cuda-11.7/lib64

권한 변경

$ sudo chmod a+r /usr/local/cuda/include/cudnn.h

$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

설치 확인

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

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

CUDA 동작 확인

Terminal 열기

$ python

import torch
torch.cuda.is_available()