Redian新闻
>
Centos 7.8 中安装 Nvidia A100 GPU驱动

Centos 7.8 中安装 Nvidia A100 GPU驱动

科技


本文介绍如何在Centos 7.8 操作系统中禁用nouveau驱动,并安装Nvidia Tesla A100型号GPU的驱动,并安装CUDA11.6和cuDNN。
系统中查看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

END

官方站点:www.linuxprobe.com

Linux命令大全:www.linuxcool.com

刘遄老师QQ:5604241

Linux技术交流群:3762708

(新群,火热加群中……)

想要学习Linux系统的读者可以点击"阅读原文"按钮来了解书籍《Linux就该这么学》,同时也非常适合专业的运维人员阅读,成为辅助您工作的高价值工具书!


微信扫码关注该文公众号作者

戳这里提交新闻线索和高质量文章给我们。
相关阅读
美国国债,破31000000000000美元!如何在 Arch Linux 中安装 Cinnamon 桌面 | Linux 中国干货:Centos8 编译安装内核CPU和GPGPU市场现状及趋势国产GPU再进一步!摩尔线程推出全新多功能GPU春晓,构建元计算新引擎NVIDIA大会上秀肌肉,却让人想到断供的GPU芯片,国产还能打吗?Tencent Said Its NFTs Weren’t an Investment. It Was Right如何在 Arch Linux 中安装 OpenOffice(新手指南) | Linux 中国说三道四(116)人老了就要“停止回忆过去”吗?关于延期举办第23届中国国际教育年会暨展览的公告 Announcement on Postponement of CACIECentOS 7 编译安装最新的 Linux Kernel 6.0 rc3每秒1000000000000000000次运算!Cerebras新超级计算机打造世界最大1350万核心AI集群如何在 Ubuntu 等 Linux 中安装 Python 3.11 | Linux 中国bāo huā shēng?bō huā shēng?CentOS 8 Stream 编译安装最新的 Linux Kernel 6.0 rc3全球最强智算来了:12000000000000000000(别数了18个0)FLOPS!虚构什么是GPU?GPU和显卡的关系?GPU国产化布局?怎样在恋爱中安全感满满?安装 Ubuntu 22.10 后要做的 10 件事 | Linux 中国Asahi Linux推出首个公开的Apple Silicon GPU驱动程序版本[梅玺阁菜话]上海老风味 蜜汁蹄脏 冰糖元蹄 冰糖肘子(视频)超越ConvNeXt!Conv2Former:用于视觉识别的Transformer风格的ConvNet如何在 Ubuntu 和其他相关 Linux 中安装 Python 3.10 | Linux 中国怎么在恋爱中安全感满满?如何在 Arch Linux 中安装 elementary OS 的 Pantheon 桌面 | Linux 中国如何在最小安装的 CentOS、RHEL、Rocky Linux 中设置互联网 | Linux 中国ACE特邀嘉宾 | Viking Global Investors-Vincent Jin |本科金融求职101一个适用于苹果芯片的原生 Linux GPU 驱动程序几乎就绪! | Linux 中国ACS Appl. Mater. Interfaces:具有自感知能力的软体驱动器Envoy Gateway会成为网关现有格局的冲击者吗?| 专访Envoy创始人以商业视角解析数据驱动,神策 2022 数据驱动大会发布全新数字化闭环产品方案|GGV Family常见骨关节疾病的中英文对照如何在 Kubuntu 22.04 中安装 KDE Plasma 5.25 | Linux 中国一个人的徒步,900公里法国之路+世界尽头:D29~招待所义工
logo
联系我们隐私协议©2024 redian.news
Redian新闻
Redian.news刊载任何文章,不代表同意其说法或描述,仅为提供更多信息,也不构成任何建议。文章信息的合法性及真实性由其作者负责,与Redian.news及其运营公司无关。欢迎投稿,如发现稿件侵权,或作者不愿在本网发表文章,请版权拥有者通知本网处理。