
引言:本文将详细介绍在较老的 CentOS 8.5.2111 服务器上安装 NVIDIA 显卡驱动和 CUDA 工具包的步骤。
对于 Ubuntu 用户,可将 yum 替换为 apt 进行安装。
安装前请务必检查 PyTorch 和 CUDA 的版本对应关系,以避免重复安装。可以在 PyTorch 官网查看 PyTorch 与 CUDA 的对应关系,而 CUDA 和 NVIDIA 驱动版本的对应关系则可以在 CUDA 12.6 Update 2 Release Notes 中找到。
安装顺序为:显卡驱动 → CUDA → CUDA Toolkit → cuDNN → PyTorch。
以服务器上的 Tesla V100 PCIe 32GB 显卡为例,适配的 PyTorch 版本与 CUDA 12.4 对应,因此我们安装 CUDA 12.4.1 及其对应的 NVIDIA 驱动 550.54.15。
首先,检查显卡型号:
lspci | grep -i nvidia
如果显卡支持 CUDA 加速,您将看到类似 NVIDIA GeForce GTX 1080 的型号。
接下来,准备安装 NVIDIA 驱动的环境:
sudo yum -y install kernel-develsudo yum -y install epel-releasesudo yum -y install gcc
如果在 CentOS 上找不到 nvidia-driver-latest-dkms,可以尝试手动安装 NVIDIA 驱动和 CUDA。
手动下载并安装 NVIDIA 驱动:
访问 NVIDIA 官方网站下载驱动,选择您的显卡型号和操作系统。下载完成后,进入下载目录,赋予安装包执行权限并运行:
wget https://us.download.nvidia.com/tesla/550.54.15/NVIDIA-Linux-x86_64-550.54.15.runsudo chmod +x NVIDIA-Linux-x86_64-*.runsudo ./NVIDIA-Linux-x86_64-*.run
在安装过程中选择默认选项,并确保禁用 nouveau 驱动(如有提示)。安装完成后,重启系统:
sudo reboot
验证 NVIDIA 驱动是否已成功安装:
nvidia-smi
若显示 NVIDIA 驱动信息,则表明安装成功。
网易人工智能
网易数帆多媒体智能生产力平台
195 查看详情
接下来,安装 CUDA:
添加 CUDA 存储库:
sudo yum-config-manager --add-repo=https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo
安装 CUDA 工具包:
sudo yum install -y libXi-devel libXmu-devel libXt-devel libXext-devel libX11-devel gcc-c++wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.runsudo sh cuda_12.4.1_550.54.15_linux.run
如果遇到临时文件空间不足的问题,可以通过设置 TMPDIR 环境变量来解决:
sudo mkdir /dev/sda1/home/tmpsudo mount --bind /dev/sda1/home/tmp /tmpsudo sh cuda_12.4.1_550.54.15_linux.runsudo umount /tmp
配置环境变量:
ls /usr/local/cuda-*echo 'export PATH=/usr/local/cuda-12.4/bin:$PATH' >> ~/.bashrcecho 'export LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH' >> ~/.bashrcsource ~/.bashrc
验证 CUDA 安装:
nvcc -V
如果正确显示 CUDA 版本信息,则表明安装成功。
结论:经过一番努力,借助网络文章和官方文档的指导,最终成功安装了 NVIDIA 驱动和 CUDA。
附注:在使用阿里云镜像时遇到问题,建议切换到清华大学镜像源。将 /etc/yum.repos.d/CentOS-Base.repo 文件中的 http://mirrors.aliyun.com 替换为 https://mirrors.tuna.tsinghua.edu.cn:
sed -e "s|^mirrorlist=|#mirrorlist=|g" -e "s|^#baseurl=http://mirror.centos.org/centos/$releasever|baseurl=https://mirrors.tuna.tsinghua.edu.cn/centos-vault/8.5.2111|g" -e "s|^#baseurl=http://mirror.centos.org/$contentdir/$releasever|baseurl=https://mirrors.tuna.tsinghua.edu.cn/centos-vault/8.5.2111|g" -i.bak /etc/yum.repos.d/CentOS-*.repo
更新缓存并安装 EPEL 仓库:
sudo yum makecachesudo yum install epel-release
如果原始仓库文件损坏,可以从阿里云镜像站下载:
repo: https://www.php.cn/link/8857f91319bfa23d6fa43b632b4e1163
epel: https://www.php.cn/link/cbc32b0f8a746c87884c5bb10a1a4289
以上就是安装 nvidia 显卡驱动及故障排查的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/819658.html
微信扫一扫
支付宝扫一扫