CentOS7 安装配置CUDA教程

前言

本篇博文介绍在CentOS7系统下安装和配置CUDA的完整流程。

一、安装准备

1.查看自己机器上的GPU

lspci | grep -i nvidia
# 若上句命令不存在，执行以下命令
yum whatprovides */lspci  
yum install pciutils

执行命令，可以查到我的GPU是Tesla P100
在这里插入图片描述

2.查看CUDA支持的GPU，包括以下几大类GPU产品
在这里插入图片描述
3.查看自己的CentOS版本

uname -m && cat /etc/redhat-release

执行命令可以查看，我的CentOS版本是7.8
在这里插入图片描述
4.CUDA支持以下linux版本

5.查看系统是否有GCC编译环境

gcc -v # 如果没有输出，就安装GCC
yum install gcc # 安装GCC

在这里插入图片描述
6.验证系统是否安装了正确的内核Headers和开发包

uname -r

没有的话可以安装

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

二、安装NVIDIA驱动

1.禁用nouveau
nouveau是一个第三方开源的Nvidia驱动，一般Linux系统安装的时候都会默认安装这个驱动。这个驱动会与nvidia官方的驱动冲突，在安装nvidia驱动和cuda之前应该先禁用nouveau
查看系统是否正在使用nouveau

lsmod | grep nouveau

如果有任何输出，那么就是nouveau在启用，需要关闭，按照以下步骤：centos7中禁用方法：

#打开如下文件
sudo vim /usr/lib/modprobe.d/dist-blacklist.conf
#写入以下内容
blacklist nouveau
options nouveau modeset=0
#保存并退出
:wq
#备份当前的镜像
sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
#建立新的镜像
sudo dracut /boot/initramfs-$(uname -r).img $(uname -r)
#重启
sudo reboot
#最后输入上面的命令验证
lsmod | grep nouveau

直到没有任何输出，说明nouveau禁用了
在这里插入图片描述
2.通过命令查找合适的驱动并安装

# 导入公钥 
rpm --import https://www.elrepo.org/RPM-GPG-KEY-elrepo.org 
# To install ELRepo for RHEL-7, SL-7 or CentOS-7: 
rpm -Uvh https://www.elrepo.org/elrepo-release-7.0-3.el7.elrepo.noarch.rpm
# 安装显卡检查程序
yum install nvidia-detect
# 查找合适驱动
nvidia-detect -v
# 安装合适版本驱动
yum install $(nvidia-detect)

3.重启（必须）

reboot

4.验证

nvidia-smi

在这里插入图片描述
3.也可以通过nvidia提供的网站，查找你的GPU适合的驱动（点击查询）
比如我的GPU是Tesla P100产品，然后查找linix 64-bit和CUDA11.0条件下的驱动

三、安装CUDA

1.下载CUDA，因为我安装的驱动版本是450.80.02，由上图可知，CUDA11.0支持该驱动，因为我下载CUDA11.0（点击下载）
在这里插入图片描述
2.安装CUDA11.0

# cuda-repo-rhel7-11-0-local-11.0.2_450.51.05-1.x86_64.rpm 是上一步下载的CUDA11.0文件
sudo rpm -i cuda-repo-rhel7-11-0-local-11.0.2_450.51.05-1.x86_64.rpm
sudo yum clean all
sudo yum install cuda

四、CUDA环境变量配置

1.运行如下命令打开profile文件

sudo vim /etc/profile

2.打开文件后在文件末尾添加路径，也就是安装目录，命令如下（若不添加这些内容，不能使用nvcc -V查看CUDA信息）：

# cuda 默认安装路径为/usr/local/cuda-11.0
export  PATH=/usr/local/cuda-11.0/bin:$PATH
export  LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64$LD_LIBRARY_PATH

3.保存后执行一个命令后重启电脑

sudo ldconfig
sudo reboot

4.查看环境变量

echo $PATH

在这里插入图片描述

5.查看CUDA版本

nvcc --version

在这里插入图片描述

五、CUDA示例执行

1.CUDA安装成功后，会有一个sample文件夹，里面有些官方示例，我们以两个示例相加为例。

cd /user/local/cuda-11.0/samples/0_Simple/vectorAdd
sudo make
./vectorAdd

出现下图，代表CUDA配置安装成功。
在这里插入图片描述

六、参考博文

CentOS 7.5 安装GPU环境（显卡驱动，CUDA，CuDNN）以及 Python3.6
Centos7.7安装Nvidia GPU驱动及CUDA以及tensorflow-GPU
Ubuntu16.04安装GPU环境（NVIDIA驱动、CUDA、CUDNN、Tensorflow-GPU安装）