[Ubuntu] Ubuntu 20.04 CUDA 11.3 install

dada·2022년 5월 20일
0

사전 작업

$ sudo apt–get install gcc
$ sudo apt-get install g++

nouveau 비활성화

STEP 1.

$ 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"

STEP 2.

$ cat /etc/modprobe.d/blacklist-nvidia-nouveau.conf
blacklist nouveau
options nouveau modeset=0

(이렇게 나오면 된거임)

STEP 3.

$ sudo update-initramfs -u

STEP 4.

$ sudo reboot

CUDA Toolkit install

STEP 1.

CUDA Toolkit 11.3 Downloads link

각 버전에 맞게 설치해주기

STEP 2.

$ wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run

STEP 3.

$ sudo reboot

reboot 후 로그인 창에서 ctrl+alt+F2 입력한 뒤
Ubuntu tty창으로 넘어가기

STEP 4.

$ ubuntu-drivers devices

recommended nvidia-driver version 확인

STEP 5.

$ sudo apt install nvidia-driver-470

470 자리에 위에서 확인한 recommended driver version 입력하기

STEP 6.

$ sudo sh cuda_11.3.0_465.19.01_linux.run

STEP 7.

  1. continue 선택
  2. accept 입력
  3. Driver은 해제하고 Install 해줌
    (CUDA Samples 아래로 있는 것들은 체크 해제해도 됨)

설치 후

STEP 1.

$ vi ~/.bashrc

STEP 2.

$ export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
$ export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

STEP 3.

$ source ~/.bashrc

STEP 4.

설치 확인

$ nvcc -V
$ nvidia-smi
profile
AI, Python 등 공부용 블로그

0개의 댓글