Ubuntu 18.04 CUDA 설치

CUDA (Computed Unified Device Architecture)

NVIDIA에서 개발한 GPU 개발 툴로 nvidia-driver, CUDA, CUDNN만 설치하면 딥러닝을 쉽게 사용할 수 있기 때문에 CUDA를 설치해보자.

앞서 nvidia-driver가 설치되어 있어야 하기 때문에 Ubuntu 18.04 Nvidia driver 설치를 보고 nvidia-driver를 설치하고 오자.

---

다운로드 및 설치

cd ~/다운로드

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

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

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb

sudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb

sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub

업데이트

sudo apt-get update

cuda-toolkit 설치

sudo apt-get install cuda-toolkit-10-2

환경변수 설정

vi ~/.bashrc

bashrc에 들어가 사진과 같이 bashrc 맨아래에 다음 두줄을 추가한다.

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

source ~/.bashrc

재부팅

reboot

Nvidia driver 설치여부 확인

nvidia driver가 설치되어 있는지 확인한다.

cat /proc/driver/nvidia/version

cat: /proc/driver/nvidia/version: No such file or directory 라고 결과가 나오면 아직 드라이버가 설치되지 않은 것이다.

다음과 같은 결과가 나와야 드라이버가 설치되어 있는 것이다.

CUDA 설치 확인

다음 명령어 실행시 다음과 같은 결과가 나오면 정상적으로 설치된 것을 확인할 수 있다.

nvcc -V

---

참고 링크

https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=deblocal 

CUDA Toolkit 10.2 Download