Terminal로 nvidia-driver, cuda, cudnn설치하기

https://freddiekim.tistory.com/47

cuda 설치 및 pytorch 설치

 

위 글을 읽기를 추천한다.  아래 글도 틀린것은 아니지만 윗 글이 그래픽 카드에따른 버전을 설명하고 있어서 

중요한 개념이라 추천함.

 

터미널로 설치하는 방법에 대해서 정리를 했다.

아래 처럼 shell로 만들어 놓으면 나중에 다시 설치할때 편할 것 같기는 하다.

꼬여 있는 경우도 있어서 0 번 완전 삭제 부분을 추가했다. 확인하고 설치하자!!!

---

0 완전 삭제

해당 파일을 만들고 메세지 나오면 

 sudo apt-get remove --purge libnvidia-compute-535

이런식으로 제거함.

#!/bin/bash

sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo apt-get autoclean
sudo rm -rf /usr/local/cuda*
sudo apt-get --purge remove 'cuda*'
sudo apt-get autoremove --purge 'cuda*'
sudo dpkg -l | grep nvidia
echo 'sudo dpkg -l | grep nvidia'
sudo dpkg -l | grep cuda
echo 'sudo dpkg -l | grep cuda'

 

1. nvidia-driver 설치

== /sys/devices/pci0000:00/0000:00:01.1/0000:01:00.0 ==
modalias : pci:v000010DEd00002684sv000010DEsd0000165Bbc03sc00i00
vendor   : NVIDIA Corporation
driver   : nvidia-driver-525 - distro non-free
driver   : nvidia-driver-535-server - distro non-free
driver   : nvidia-driver-535-open - distro non-free
driver   : nvidia-driver-525-server - distro non-free
driver   : nvidia-driver-525-open - distro non-free
driver   : nvidia-driver-535-server-open - distro non-free recommended
driver   : nvidia-driver-535 - distro non-free
driver   : xserver-xorg-video-nouveau - distro free builtin

lspci | grep -e VGA
ubuntu-drivers devices
sudo apt install nvidia-driver-525

 

2. cuda 설치

https://developer.nvidia.com/cuda-toolkit-archive 

위 사이트 들어가서 적당한 버전을 찾는다. 내 경우 12버전을 골랐다.

만일 버전이 너무 높아서 문제가 생기면 처음 부터 다시 시작하거나 bashrc에서 버전 정보를 변경해주면 된다.

#!/bin/bash

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda-repo-ubuntu2204-12-0-local_12.0.0-525.60.13-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-12-0-local_12.0.0-525.60.13-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-12-0-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

nvcc -V를 쳐서 아무것도 나오지 않으면

sudo apt-get install nvidia-cuda-toolkit

를 쳐서 설치함.

3. bashrc 설정

export PATH="/usr/local/cuda-12.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-12.0/lib64:$LD_LIBRARY_PATH"

 

4. cudnn 설치

설치 위치 https://developer.nvidia.com/rdp/cudnn-archive위 사이트에서 cuda에 맞는 버전을 선택하면 됨.나의 경우는 cuda 12.0 에  cuDnn 8.1.3버전을 선택함.

https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html

여기에 설치 방법이 쓰여져 있지만 약간 복잡함.

총 3가지 방법이 있는데 여기서 전 

1.3.2. Debian Local Installation

Installation Guide - NVIDIA Docs

해당 방법을 사용했습니다. 

예제와 내가 사용한 명령어가 달랐다. 아마도 문서가 업데이트 되지 않은듯하다.

Tab을 눌러서 어떤 명령어가 가능하지 확인해보면 된다.

나의 경우는 아래와 같았다.

#!/bin/bash
sudo dpkg -i cudnn-local-repo-ubuntu2204-8.8.1.3_1.0-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-12-0-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install nvidia-cudnn

# Install the runtime library.
sudo apt-get install libcudnn8=8.8.1.3-1+cuda12.0 
# Install the developer library
sudo apt-get install libcudnn8-dev=8.8.1.3-1+cuda12.0 
# Install the code samples. 
sudo apt-get install libcudnn8-samples=8.8.1.3-1+cuda12.0

 

출처 : https://linuxconfig.org/how-to-install-the-nvidia-drivers-on-ubuntu-20-04-focal-fossa-linux

출처 : https://settembre.tistory.com/447

출처 : https://forums.developer.nvidia.com/t/freeimage-is-not-set-up-correctly-please-ensure-freeimae-is-set-up-correctly/66950/2

 

5. 검증

#!/bin/bash

# 예제를 실행하는데 필요한 라이브러임.
sudo apt-get install libfreeimage3 libfreeimage-dev

cp -r /usr/src/cudnn_samples_v8/ $HOME\

cd  $HOME/cudnn_samples_v8/mnistCUDNN

make clean && make

./mnistCUDNN

# Test passed! 을 볼 수 있면 제대로 설치가 된것임.

이미지를 이용하기 때문에 이미지 라이브러를 설치해야함.