CUDA 설치 우분투 환경 - jinwooklim/my-exp GitHub Wiki

참조 : https://hiseon.me/linux/ubuntu/cuda-install/

CUDA 설치 우분투 환경

  1. 먼저 NVIDIA 드라이버가 설치되었는지 확인

    $ nvidia-smi

  2. CUDA를 설치 하기 이전에, 설치하고자 하는 CUDA 버전이 현재 GPU의 Compute Capability(계산 능력)을 지원하는지 확인

    1. https://developer.nvidia.com/cuda-gpus
    2. https://en.wikipedia.org/wiki/CUDA*
CUDA Compute capability Architecture
6.5 Last version with support for compute capability 1.x (Tesla)
7.5 support for compute capability 2.0 – 5.x (Fermi, Kepler, Maxwell)
8.0 support for compute capability 2.0 – 6.x (Fermi, Kepler, Maxwell, Pascal), last version with support for Compute Capability 2.x (Fermi)
9.0 – 9.2 support for compute capability 3.0 – 7.2 (Kepler, Maxwell, Pascal, Volta)
10.0 – 10.1 support for compute capability 3.0 – 7.5 (Kepler, Maxwell, Pascal, Volta, Turing)

패키지 리스트 추가

  1. 우분투 배포판 버전을 확인

    $ release="ubuntu"$(lsb_release -sr | sed -e "s/\.//g")

    $ echo $release

  2. Repository 추가

    $ sudo apt install sudo gnupg

    $ sudo apt-key adv --fetch-keys "http://developer.download.nvidia.com/compute/cuda/repos/"$release"/x86_64/7fa2af80.pub"

    $ sudo sh -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/'$release'/x86_64 /" > /etc/apt/sources.list.d/nvidia-cuda.list'

    $ sudo sh -c 'echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/'$release'/x86_64 /" > /etc/apt/sources.list.d/nvidia-machine-learning.list'

    $ sudo apt update

CUDA 설치

  1. cuda 10.0을 설치 (example)

    $ sudo apt-get install cuda-10-0

  2. cudnn 7.x 설치 (example)

    $ sudo apt-get install libcudnn7-dev

버전 확인

  1. cuda

$ cat /usr/local/cuda/version.txt

CUDA Version 10.0.130

  1. cudnn

$ cat /usr/include/cudnn.h | grep -E "CUDNN_MAJOR|CUDNN_MINOR|CUDNN_PATCHLEVEL"

CUDNN 8.x 부터 변경되었다
$ cat /usr/include/cudnn_version.h | grep -E "CUDNN_MAJOR|CUDNN_MINOR|CUDNN_PATCHLEVEL"

#define CUDNN_MAJOR 7

#define CUDNN_MINOR 5

#define CUDNN_PATCHLEVEL 1

#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

$ sudo find / -name libcudnn*.*

/usr/lib/x86_64-linux-gnu/libcudnn.so.7

/usr/lib/x86_64-linux-gnu/libcudnn.so.7.5.1

/var/lib/dpkg/info/libcudnn7.list

/var/lib/dpkg/info/libcudnn7.md5sums

/var/lib/dpkg/info/libcudnn7.triggers

/var/lib/dpkg/info/libcudnn7.shlibs