Ubuntu 환경에서 CUDA & cuDNN 설치

Ubuntu에서 NVIDIA Driver를 설치하기

이글은 인공지능(머신러닝)을 위해서 NVIDIA CUDA를 설치하는 과정을 정리한 것입니다.

  • 사용환경
  • ubuntu 14.04.05
  • ubuntu 16.04.01

리눅스 드라이버 제거

/etc/modprobe.d/blacklist-nouveau.conf 를 각자 사용하는 편집기로 다음의 항목을 추가합니다.

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

blacklist nouveau
options nouveau modeset=0
  • 업데이트 합니다.

sudo update-initramfs -u

CUDA 8.0 다운로드

wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_8.0.61-1_amd64.deb

  • 16.04.x cud 8.0 다운로드

wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb

다운받은 CUDA 8.0 을 실행

  • cuda 8.0을 저장소 정보에 추가 (ubuntu 14.04)

sudo dpkg -i cuda-repo-ubuntu1404-8-0-local_8.0.61-1_amd64.deb

  • cuda 8.0을 저장소 정보에 추가 (ubuntu 16.04)

sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb

  • cuda 8.0을 설치
sudo apt-get update
sudo apt-get install cuda
  • cuda_8.0.44_linux-run 를 받아서 자동 실행에 따라서 설치하는 다른 방법

sudo apt-get install build-essential chmod +x cuda_8.0.44_linux-run ./cuda_8.0.44_linux-run

설치하고 나서 리부팅하고, 그리고나서 해야 할 것들.

  • 쉘의 환경 설정 파일(~/.bashrc)에 CUDA 관련 PATH 를 추가합니다.

echo “export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}” >> ~/.bashrc

  • 쉘의 환경 설정 파일(~/.bashrc)에 CUDA 관련 LIBRARY PATH를 추가합니다.

echo “export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64\ ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}” >>~/.bashrc

CUDA 설치 확인

cat /proc/driver/nvidia/version

NVRM version: NVIDIA UNIX x86_64 Kernel Module 367.48 Sat Sep 3 18:21:08 PDT 2016

  • 2016년 2월 13일 (update)

$ nvidia-smi

Mon Feb 13 18:08:24 2017       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 367.48                 Driver Version: 367.48                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX TIT...  Off  | 0000:01:00.0     Off |                  N/A |
|  0%   55C    P0    58W / 250W |      0MiB / 12198MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

cuDNN 설치

  • https://developer.nvidia.com 에 접속해서 회원 가입과 약관을 동의하여 cudnn을 받는다.

  • scp 등을 이용해서 linux 로 복사한다.

  • 압축을 풀고, 복사한다.

tar xvd cudnn-8.0-linux-x64-v5.1.tgz sudo cp cuda/lib64/* /usr/local/cuda/lib64/ sudo cp cuda/include/* /usr/local/cuda/include/ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

예제 코드 실행

cuda-install-samples-8.0.sh ~ cd ~/NVIDIA_CUDA-8.0_Samples/5_Simulations/nbody $ make ./nbody