本文环境 CentOS7
1 nvidia显卡驱动
通过ELRepo源的nvidia-detect命令来自动化安装驱动，省时省力，告别传统手工操作，一步一个坑
1.1 安装ELRepo镜像源
1.导入公共秘钥
rpm --import https://www.elrepo.org/RPM-GPG-KEY-elrepo.org
2.选择对应版本安装
#centos7
rpm -Uvh http://www.elrepo.org/elrepo-release-7.0-2.el7.elrepo.noarch.rpm
#centos6
rpm -Uvh http://www.elrepo.org/elrepo-release-6-6.el6.elrepo.noarch.rpm
#centos5
rpm -Uvh http://www.elrepo.org/elrepo-release-5-5.el5.elrepo.noarch.rpm
1.2 查找需要安装的驱动版本
1.安装nvidia-detect
通过nvidia-detect这个命令来查找我们需要安装的驱动版本,所以我们需要来安装它
yum -y install gcc kernel-devel kernel-headers
yum install -y nvidia-detect
2.运行nvidia-detect
nvidia-detect
输出以下结果，说明安装最新版本就可以，不需要制定版本
1.3 禁用nouveau
1.判断是否开启nouveau
lsmod | grep nouveau
2.关闭nouveau(上方命令如果无输出则无需进行此步)
sudo echo "blacklist nouveau" > /etc/modprobe.d/blacklist-nouveau.conf
sudo echo "blacklist nouveau" > /lib/modprobe.d/dist-blacklist.conf
sudo echo "options nouveau modeset=0" > /etc/modprobe.d/blacklist-nouveau.conf
sudo echo "options nouveau modeset=0" > /lib/modprobe.d/dist-blacklist.conf
sudo dracut --force
1.4 安装nvidia显卡驱动
此处根据上方的nvidia-detect命令输出的第一行进行安装
sudo yum install -y kmod-nvidia.x86_64
1.5 重启后测试
1.重启
reboot
2.测试
运行下方命令，如果出现显卡详情则代表安装成功
nvidia-smi
2 cuda
2.1 下载cuda
根据上面安装的驱动版本安装cuda，如果上方是kmod-nvidia，则可以安装最新版本，如果是其他的则需要根据下图进行选择版本下载
注意：显卡算力在3.0以下的老显卡，需要安装cuda8，cuda9会导致一些地方不可用 nvidia显卡算力表
选择配置，但是Installer Type必须选择runfile,根据下方给出的url下载安装包，可选择使用wget直接在centos下载，也可以选择使用其他方式下载然后上传至centos
2.2 安装cuda
在ubuntu上运行下方代码
sh cuda_10.0.130_410.48_linux.run
根据下方步骤进行安装
- 文档，输入
q - 接受，输入
accept，回车 - 是否安装驱动，输入
n，回车 - 是否安装cuda，输入
y，回车 - 默认安装目录，
回车 - 是否建立目录软链接，
回车 - 是否安装默认程序，输入
y - 默认程序位置，
回车
2.3 配置cuda
vim /etc/profile
加入下面三行环境变量
export CUDA_HOME=/usr/local/cuda:/usr/local/cuda-10.0
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda-10.0/lib64
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/root/TensorRT-6.0.1.5/lib:/usr/local/cuda/extras/CUPTI/lib64
export PATH=$PATH:$CUDA_HOME/bin
保存
source /etc/profile
3 cudnn
3.1 下载cudnn
cudnn官网,需要登录下载
选择跟cuda相对应的版本，选择cuDNN Library for Linux版本下载
3.2 安装cudnn
3.2.1 解压
tar -zxvf cudnn-10.0-linux-x64-v7.6.5.32.tgz
3.2.2 安装
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
3.3 查看版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
