Ubuntu 20.04 CUDA 재설치

 

 

1. 서론

cuda를 설치하는 것이 은근히 오래 걸리고 어렵기 때문에 정리하고자 한다.

 

2. 

2.1 CUDA 삭제

sudo apt-get --purge -y remove 'cuda*'
sudo apt-get --purge -y remove 'nvidia*'
sudo apt-get autoremove --purge cuda

# cudnn remove
cd /usr/local/
sudo rm -rf cuda*

~/.bashrc나 /etc/profile에 추가되어있는 CUDA 관련 설정도 제거해야 한다.

export PATH=$PATH:/usr/local/cuda-11.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-{version}/lib64
export CUDADIR=/usr/local/cuda-{version}

2.2 nouveau 비활성화

1. Nvidia nouveau driver 블랙리스트 설정

터미널을 열고 아래의 명령어 입력

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"

2. 비활성화 확인

아래 cat 명령어 입력하고 다음과 같이 출력이 나오는지 확인

cat /etc/modprobe.d/blacklist-nvidia-nouveau.conf

blacklist nouveau
options nouveau modeset=0

리부트

sudo reboot

3. 커널 업데이트

sudo update-initramfs -u

2.3 Nvidia driver + CUDA 설치

Nvidia 드라이버 설치 과정이 잘 정리된 블로그이다. CUDA를 설치하기 전에 Nvidia driver먼저 설치하고 nvidia-smi를 터미널에 입력하여 설치됐는지 확인한다. 여기서 나오는 CUDA Version은 설치된 CUDA Version이 아니고 설치 가능한 CUDA 최신 버전이 출력

https://pstudio411.tistory.com/entry/Ubuntu-2004-Nvidia%EB%93%9C%EB%9D%BC%EC%9D%B4%EB%B2%84-%EC%84%A4%EC%B9%98%ED%95%98%EA%B8%B0

[Ubuntu 20.04 LTS]Nvidia드라이버 설치하기

아래의 홈페이지에 접속한 후 원하는 CUDA 버전을 선택한다.

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

CUDA Toolkit Archive

아래 사진과 같이 선택

Installer Type에서 deb를 선택할 경우 CUDA 최신 버전이 설치된다. runfile을 선택해서 11.3을 설치하려 했지만 설치가 잘 진해되지 않아 deb로 설치하였다.

 

위의 사진처럼 클릭하면 아래와 같은 명령어가 나온다. 순서대로 복붙 하여 cuda를 설치한다.

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/
sudo apt-get update
sudo apt-get -y install cuda

3. 설치 확인

pytorch를 설치한 경우 터미널을 열고 python을 실행하여 torch 라이브러리로 cuda가 사용 가능한지 확인해본다.