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
