CUDA 설치하기

우분투 버전 확인

# apt install -y lsb-core
release="ubuntu"$(lsb_release -sr | sed -e "s/\.//g")
echo $release

CUDA 패키지 리스트 추가

sudo apt install gnupg
sudo apt-key adv --fetch-keys "http://developer.download.nvidia.com/compute/cuda/repos/"$release"/x86_64/7fa2af80.pub"
sudo sh -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/'$release'/x86_64 /" > /etc/apt/sources.list.d/nvidia-cuda.list'
sudo sh -c 'echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/'$release'/x86_64 /" > /etc/apt/sources.list.d/nvidia-machine-learning.list'
sudo apt update

# GPG pubkey 관련 에러 뜨면 아래 명령 실행
sudo apt-key adv --keyserver keyserver.ubuntu.com --recv-keys A4B469963BF863CC

CUDA & CuDNN 설치 (버전 확인)

# sudo apt-get install cuda-11-7  # 이걸로 하면 드라이버까지 설치
sudo apt install cuda-toolkit-11-7
sudo apt-get install libcudnn8-dev

환경변수 추가

vim ~/.bashrc
>> export PATH=/usr/local/cuda/bin:$PATH
>> export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

재부팅

sudo reboot now