驱动安装流程

nvidia驱动安装

1. 屏蔽系统自带驱动

#查看几个显卡
sudo lspci |grep -i nvidia

# 屏蔽系统驱动
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
########################
blacklist nouveau
options nouveau modeset=0
###################

# 创建/etc/modprobe.d/nouveau-kms.conf文件，将options nouveau mdeset=0添加进去：
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf

#更新一下initramfs：
sudo update-initramfs -u

#多个内核的时候 更新到所有内核
sudo update-initramfs -u -k all

#重启机器检查自带的模块是否已屏蔽
reboot
sudo lsmod | grep nouveau

2. 安装cudn

sudo apt-get update
sudo apt-get install build-essential
xz@xz128:~/local/gpu$ pwd
/home/xz/local/gpu

bash cuda_12.5.1_555.42.06_linux.run

# .bashrc下填入以下内容
vim ~/.bashrc
###########################
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-12.3/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
########################

# 创建nvcc软链接
sudo ln -s /usr/local/cuda/bin/nvcc /usr/bin/nvcc
#使用nvcc命令查看cuda的版本：
nvcc --version

3. 安装cudnn

# 解压文件到/root下
 tar xvf cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz -C /root

# 复制文件到cuda目录
 sudo cp -p /root/cudnn-linux-x86_64-8.9.7.29_cuda12-archive/include/cudnn*.h /usr/local/cuda/include/
 sudo cp -p /root/cudnn-linux-x86_64-8.9.7.29_cuda12-archive/lib/libcudnn* /usr/local/cuda/lib64/

4. 验证驱动

nvidia-smi