CentOS7 安装配置CUDA教程
发布日期:2021-05-28 17:02:47 浏览次数:8 分类:技术文章

本文共 2202 字,大约阅读时间需要 7 分钟。

CentOS7 安装配置CUDA教程

前言

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

一、安装准备

1.查看自己机器上的GPU

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

执行命令,可以查到我的GPU是Tesla P100

在这里插入图片描述

2.,包括以下几大类GPU产品

在这里插入图片描述
3.查看自己的CentOS版本

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

执行命令可以查看,我的CentOS版本是7.8

在这里插入图片描述
4.
在这里插入图片描述
5.查看系统是否有GCC编译环境

gcc -v # 如果没有输出,就安装GCCyum 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 nouveauoptions 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.rpmsudo yum clean allsudo yum install cuda

四、CUDA环境变量配置

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

sudo vim /etc/profile

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

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

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

sudo ldconfigsudo reboot

4.查看环境变量

echo $PATH

在这里插入图片描述

5.查看CUDA版本

nvcc --version

在这里插入图片描述

五、CUDA示例执行

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

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

出现下图,代表CUDA配置安装成功。

在这里插入图片描述

六、参考博文

转载地址:https://blog.csdn.net/qq_34720818/article/details/112710840 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!

上一篇:Linux常用操作命令
下一篇:CUDA编程入门教程

发表评论

最新留言

逛到本站,mark一下
[***.202.152.39]2024年02月18日 10时19分19秒