前提

验证是否存在支持CUDA的GPU

$ lspci | grep -i nvidia
00:08.0 3D controller: NVIDIA Corporation GP100GL (rev a1)

验证支持的Linux版本

$ uname -m && cat /etc/*release

x86_64
CentOS Linux release 7.3.1611 (Core)

验证安装的gcc

$ gcc --version

gcc (GCC) 4.8.5 20150623 (Red Hat 4.8.5-16)

验证kernel版本

$ uname -r

3.10.0-514.26.2.el7.x86_64

安装kernel headers及开发包

$ sudo yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)

安装cuda

安装yum源

sudo rpm -i cuda-repo-rhel7-9-1-local-9.1.85-1.x86_64.rpm

清理yum源

sudo yum clean all

安装cuda

sudo yum install cuda

在/etc/profile.d/cuda.sh配置cuda

# cuda
export PATH=/usr/local/cuda/bin:$PATH

激活配置

source /etc/profile

进入cuda安装目录，编译sample

cd /usr/local/cuda/samples
make

cuDNN

cuDNN（NVIDIA CUDA Deep Neural Network）是深度神经网络的GPU加速库，是NVIDIA Deep Learning SDK的一部分。