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
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.6Centos7.7安装Nvidia GPU驱动及CUDA以及tensorflow-GPUUbuntu16.04安装GPU环境(NVIDIA驱动、CUDA、CUDNN、Tensorflow-GPU安装)