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 条评论