Ubuntu 16.04 CUDA 설치하기

Ubuntu 16.04 환경에서 CUDA 설치

사전 준비

nvidia driver

https://oboki.net/workspace/system/linux/ubuntu-nvidia-graphic-driver-%EC%84%A4%EC%B9%98/

Download CUDA local installer(.run)

https://developer.nvidia.com/cuda-downloads

cuda_8.0.61_375.26_linux-run

Remove any other installation

#sudo apt-get purge nvidia-cuda*
#sudo apt-get purge nvidia-cuda*
#sudo apt-get purge nvidia-*

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

sudo rm -rf /usr/local/cuda-*
sudo rm -rf /usr/local/cuda

설치

Logout from GUI and change boot mode to Command-line prompts

Press [CTRL] + [ALT] + [F2]

콘솔에서 진행하는 경우 command-line 환경으로 변경해준다. ssh 터미널로 접속해서 설치하는 경우에는 그냥 진행하면 됨.

Stop lightdm

sudo service lightdm stop

blacklist-nouveau.conf 파일 생성

sudo vi /etc/modprobe.d/blacklist-nouveau.conf

blacklist nouveau
options nouveau modeset=0

Update initramfs

sudo update-initramfs -u

Run installer

sudo ./cuda_8.0.61_375.26_linux-run --override

환경 복구

blacklist-nouveau.conf 파일 수정

sudo vi /etc/modprobe.d/blacklist-nouveau.conf

#blacklist nouveau
#options nouveau modeset=0

Update initramfs

sudo update-initramfs -u

Start lightdm

sudo service lightdm start

reboot

sudo reboot now

Install CUDA package

sudo apt-get install cuda

설치 확인

/usr/local/cuda 경로에 바이너리와 라이브러리가 생성돼 있는지 확인하고, 아래 명령으로 그래픽 드라이버와 nvcc가 잘 동작하는지 확인한다.

nvcc --version

nvidia-smi

CUDA Sample 실행 테스트