Centos 7.8 中安装Nvidia A100 GPU驱动

系统中查看GPU信息

系统安装完成之后,进入系统,使用lspci 命令查询一下GPU是否存在、型号信息是什么。

[root@node01 ~]# lspci | grep -i nvidia
下载NVIDIA Tesla A100驱动

通过lspci查询到GPU的型号之后,访问官网nvidia.com下载驱动程序可以按照如下选择,选择产品类型、系列、型号、然后根据自己的操作系统来选择。
注意:如果操作系统是Linux,尽量选择Linux 32-bit/Linux 64-bit,不需要选择详细的Linux发行版本。测试是发现选择详细的Linux发行版本,安装驱动之后,找不到nvidia-smi命令。

安装gcc等依赖包

当安装GPU驱动时,提示缺少相关的依赖包,在此,我们需要提前安装相关的依赖包,目前需要用到的是gcc , g++ , make :

[root@localhost ~]# yum -y install gcc gcc-c++ kernel-devel make
屏蔽nouveau开源版本的GPU驱动

当系统安装完成之后,会安装系统开源的NVIDIA驱动版本,名称为nouveau。查看nouveau驱动是否加载。

[root@localhost ~]# lsmod |grep -i nouveau


下面将屏蔽该驱动。编辑/etc/default/grub文件,修改GRUB_CMDLINE_LINUX一行:

[root@localhost ~]# cat /etc/default/grub


保存退出,更新grub:

[root@localhost ~]# grub2-mkconfig -o /boot/grub2/grub.cfg
Generating grub configuration file ...
Found linux image: /boot/vmlinuz-3.10.0-1160.el7.x86_64
Found initrd image: /boot/initramfs-3.10.0-1160.el7.x86_64.img
Found linux image: /boot/vmlinuz-0-rescue-91fd57f1c90a4a2da20ef93ed7f886c6
Found initrd image: /boot/initramfs-0-rescue-91fd57f1c90a4a2da20ef93ed7f886c6.img
done
[root@localhost ~]# grub2-mkconfig -o /boot/grub2/grub.efi.cfg
Generating grub configuration file ...
Found linux image: /boot/vmlinuz-3.10.0-1160.el7.x86_64
Found initrd image: /boot/initramfs-3.10.0-1160.el7.x86_64.img
Found linux image: /boot/vmlinuz-0-rescue-91fd57f1c90a4a2da20ef93ed7f886c6
Found initrd image: /boot/initramfs-0-rescue-91fd57f1c90a4a2da20ef93ed7f886c6.img


编辑/lib/modprobe.d/dist-blacklist.conf,把 blacklist nvidiafb 改成 #blacklist nvidiafb,然后在最下面添加blacklist nouveau和。

[root@localhost ~]# vim /lib/modprobe.d/dist-blacklist.conf
# blacklist nvidiafb
blacklist nouveau
options nouveau modeset=0



备份initrams并使用dracut重新建立initramfs并重启系统:

[root@localhost ~]# mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
[root@localhost ~]# dracut -v /boot/initramfs-$(uname -r).img $(uname -r)
[root@localhost ~]#  reboot

查看nouveau模块是否加载,不显示的话就表示已经禁用:

# lsmod | grep nouveau

安装NVIDIA GPU驱动

安装下载的GPU驱动:NVIDIA-Linux-x86_64-510.47.03.run ,目前驱动版本为:510.47.03,如下执行该驱动文件,即可安装。

[root@localhost ~]# ./NVIDIA-Linux-x86_64-346.47.run



输入nvidia-smi来查看是否可以看到GPU信息:

[root@localhost ~]# nvidia-smi


至此安装完成。

安装CUDA 11.6

访问nvidia官网,下载CUDA,cuda的链接为:https://developer.nvidia.com/cuda-downloads
选择runfile文件来安装。

[root@localhost ~]# wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run
[root@localhost ~]# sudo sh cuda_11.6.0_510.39.01_linux.run

如下图,Driver选项不要勾选了,前面已经安装GPU驱动了。

创建环境变量,编辑~/.bashrc文件:

[root@localhost ~]# vim ~/.bashrc

将下面命令追加到文件最后面:

export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

nvcc命令创建一个软连接到/usr/bin目录:

[root@localhost ~]# sudo ln -s /usr/local/cuda/bin/nvcc /usr/bin/nvcc

使用nvcc命令查看cuda的版本:

[root@localhost ~]# nvcc --version

本文原创地址:https://www.linuxprobe.com/centos7-a100-gpu.html编辑:逄增宝,审核员:逄增宝

为您推荐一些与本文相关的文章:

  • 《 SQL Server从入门到精通》pdf电子书免费下载
  • Java 组合模式及其应用
  • 设置 SSH 隧道(端口转发)实战
  • EX440:红帽认证消息传递管理专家考试
  • 英国网络安全中心处理已发现漏洞的方法
  • 客官留步,让我们走进CxO 们的容器实践指南
  • 《移动互联网之路》pdf电子书免费下载
  • 方便你我他的四个开源调查工具
  • Apache服务器常规操作
  • 号称最美Linux发行版—elementary OS 5.1 “Hera”发布
以上是Centos 7.8 中安装Nvidia A100 GPU驱动的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>