majiangwei
作者majiangwei2021-11-17 10:53
售前技术支持, 广州市品高软件股份有限公司

CentOS 7.6安装 NVIDIA 显卡驱动 、cuda以及gpu_burn并测试

字数 1089阅读 3380评论 0赞 2

1. 安装依赖环境:

yum install kernel-devel gcc -y

2. 检查内核版本和源码版本,保证一致

ls /boot | grep vmlinu rpm -aq | grep kernel-devel

3. 屏蔽系统自带的 nouveau

查看命令: lsmod | grep nouveau 修改 dist-blacklist.conf 文件: vim /lib/modprobe.d/dist-blacklist.conf 将 nvidiafb 注释掉 : #blacklist nvidiafb 然后添加以下语句:

blacklist nouveau

options nouveau modeset=0

屏蔽之前

屏蔽之后

4. 重建 initramfs image 步骤 (设置完 reboot ,再查看)

mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak

dracut /boot/initramfs-$(uname -r).img $(uname -r)

5. 修改运行级别为文本模式

systemctl set-default multi-user.target

6 、设置好之后安装下载的驱动:

chmod +x NVIDIA-Linux-x86_64-440.64.run ./NVIDIA-Linux-x86_64-440.64.run

7 、安装驱动

执行 nvidia-smi

8 、驱动安装完成,安装 cuda

输入 accept

选择 install ,如果已安装驱动,可以驱动这里不选择,按空格

9 、安装完添加 ~ /.bashrc

export PATH="$PATH:/usr/local/cuda-11.0/bin"

export LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64"

安装完查看版本

10 、安装 gpu_burn 测试工具

下载 gpu_burn ,解压安装

修改 Makefile 文件

vim 编辑 Makefile 文件,去掉 -arch=compute_30

给 Makefile 添加执行权限

chmod +x Makefile

make install 安装

生成 gpu_burn 脚本 ,执行 ./gpu_burn 即可测试 GPU

扩展:

多块 GPU 卡的情况可以指定测试某张卡 , 比如指定 0 和 1 号卡

export CUDA_VISIBLE_DEVICES=0,1

./gpu_burn 1000

如果觉得我的文章对您有用,请点赞。您的支持将鼓励我继续创作!

2

添加新评论0 条评论

Ctrl+Enter 发表

相关文章

相关问题

相关资料

X社区推广