[Ubuntu 18.04] CUDA 10.2 설치하기

1. 그래픽 카드 정보 확인

먼저 자신의 컴퓨터의 그래픽 카드 확인해본다.

*터미널에서 복사 및 붙여넣기는 Ctrl+Shift+C, Ctrl+Shift+V

$ ubuntu-drivers devices

Titan xp 인것을 확인 CUDA-Wiki에 들어가서 내 그래픽 카드와 호환되는 쿠다 버전을 확인해본다

(호환되지 않으면 정상적으로 설치되지 않을 확률이 크다)

https://en.wikipedia.org/wiki/CUDA

CUDA - Wikipedia

Titan xp는 CUDA SDK Version(s) 중 Pascal에 해당된다. 해당 목록을 클릭하면 모델명이 나와 있으므로 참고하기! Pascal은 CUDA 8.0부터 12.4부터 지원하는데 난 지금 10.2가 필요하기 때문에 10.2를 설치해주도록 하겠다

 

2. CUDA 10.2 Download

https://developer.nvidia.com/cuda-10.2-download-archive

CUDA Toolkit 10.2 Download

운영체제와 버전을 선택하고 설치해준다. 난 그냥 cluster(local)로 설치하였는데 뭐가 다른지는 정확히 모름,, 알려주세요

 

3. 터미널에 접속하여 쿠다 설치하기

터미널에 접속한뒤 아래의 명령을 차례대로 입력해준다

$ 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

업데이트 해주기

$ sudo apt-get -y install cuda-10-2

해당 명령어를 입력하여 cuda 10.2 설치

 

4. 환경변수 편집

$ sudo vi ~/.bashrc

설치가 끝났다면 환경변수 설정을 위해 명령어를 입력해준다

 

그럼 이런 창이 뜬다.

 

export PATH=$PATH:/usr/local/cuda-10.2/bin
export CUDADIR=/usr/local/cuda-10.2
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64

맨 아래로 가서 이 세줄을 추가해준다 이후 Esc 누르고 :wq 입력하여 저장한다.

 

$ source ~/.bashrc

환경 변수 불러오기 

 

이후 재부팅을 해준다

 

$ nvidia-smi

재부팅 이후 해당 명령어를 입력하면?

설치가 완료된것을 확인해 볼 수 있다