CentOS部署显卡驱动:CUDA,cuDNN

1 环境

CentOS

2 查看显卡安装

lspci | grep -i nvidia

3b:00.0 3D controller: NVIDIA Corporation GP104GL [Tesla P4] (rev a1)
5e:00.0 3D controller: NVIDIA Corporation GP104GL [Tesla P4] (rev a1)
86:00.0 3D controller: NVIDIA Corporation GP104GL [Tesla P4] (rev a1)

3 安装驱动程序

3.1 下载驱动

传送门:https://www.nvidia.cn/Download/index.aspx?lang=cn
选择对应的显卡版本.

图 显卡驱动配置

3.2 安装驱动

下载文件:NVIDIA-Linux-x86_64-418.67.run

3.2.1 准备操作

• 修改权限
  添加可运行.

sudo chomd +x NVIDIA-Linux-x86_64-418.67.run

• 退出GUI界面
  退出X Server,才能安装驱动.

sudo systemctl stop gdm.service

# 停止
sudo systemctl stop gdm.service
# 查看状态
sudo systemctl status gdm.service
# 启动
sudo systemctl start gdm.service

• 关闭Nouveau

cd /usr/lib/modprobe.d
sudo vim dist-blacklist.conf

blacklist nouveau
options nouveau modeset=0

• 查看nouveau状态

lsmod | grep nouveau

• 备份镜像

sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r)-nouveau.img

• 新建镜像

sudo dracut /boot/initramfs-$(uname -r).img $(uname -r)

• 内核版本

cd /usr/src/kernels
ls

• 完整内核版本

3.10.0-957.27.2.el7.x86_64

使用uname -r获取的内核版本不全.

3.2.2 安装插件

• 安装gcc
  编译C.

sudo yum install -y gcc

• 安装gcc++
  编译C++

sudo yum install -y gcc-c++

• 安装kernel-devel
  内核依赖

sudo yum install -y kernel-devel

• 安装kernel-headers
  内核头文件

sudo yum install -y kernel-headers

• 安装驱动

sudo ./NVIDIA-Linux-x86_64-418.67.run --no-opengl-files --kernel-source-path=/usr/src/kernels/3.10.0-957.27.2.el7.x86_64

使用root安装,不安装opengl,因为安装opengl,CentOS界面UI不能正常启动,添加内核路径.

3.2.3 查看显卡信息

nvidia-smi

图 显卡信息

4 安装CUDA

4.1 下载

传送门:https://developer.nvidia.com/cuda-toolkit-archive
CUDA8.0

4.2 安装

• 添加可执行权限

sudo chmod +x cuda_8.0.44_linux.run

• 安装CUDA

sudo ./cuda_8.0.44_linux.run --no-opengl-libs

• 跳过阅读许可

q

• 同意

accept

• 不安装Nvidia加速
  这个加速驱动是旧版本,已经安装了最新的驱动,所以这个不安装,选择n.

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 367.48?
(y)es/(n)o/(q)uit: n

• 安装Toolkit

Install the CUDA 8.0 Toolkit?
(y)es/(n)o/(q)uit: y

• 默认安装目录

Enter Toolkit Location
 [ default is /usr/local/cuda-8.0 ]:

• 安装symbolic

Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y

• 卸载CUDA

yum uninstall /usr/local/cuda-8.0/bin

4.3 配置路径

• 配置

sudo vim ~/.bashrc

export CUDA_HOME=/usr/local/cuda-8.0
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64

• 生效

source ~/.bashrc

4.4 查看CUDA版本

• 详细信息

nvcc -V

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Sun_Sep__4_22:14:01_CDT_2016
Cuda compilation tools, release 8.0, V8.0.44

• 版本信息

cat /usr/local/cuda/version.txt

CUDA Version 8.0.44

5 安装cuDNN

5.1 下载

传送门:https://developer.nvidia.com/rdp/cudnn-archive
与CUDA版本对应:cuDNN6.0

5.2 解压

CUDA安装目录:/usr/local/cuda和/usr/local/cuda-8.0

tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz -C /usr/local

直接将添加的文件解压到local目录下的cuda和cuda-8.0文件夹.

5.3 查看cuDNN信息

cat /usr/local/cuda-8.0/include/cudnn.h | grep CUDNN_MAJOR -A 2

#define CUDNN_MAJOR      6
#define CUDNN_MINOR      0
#define CUDNN_PATCHLEVEL 21
--
#define CUDNN_VERSION    (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

---

[参考文献]
[1]https://www.jianshu.com/p/a201b91b3d96
[2]https://blog.csdn.net/qq_41493990/article/details/82183035
[3]https://blog.csdn.net/fortuna_i/article/details/83141070
[4]https://www.jianshu.com/p/36dcf5185f01
[5]https://blog.csdn.net/zhe_csdn/article/details/88427829
[6]https://www.linuxidc.com/Linux/2017-01/139319.htm
[7]https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html