yolov5环境配置搭建-----nvidia驱动+cuda+cudnn安装

1. 安装Nvidia 显卡驱动

步骤一： 安装更新软件列表和依赖项

在安装Nvidia显卡驱动之前，需要更新软件列表和必要的依赖项。

sudo apt-get update		# 更新软件列表

sudo apt-get install g++		# 下载g++编译器

sudo apt-get install gcc		# 下载gcc编译器

sudo apt-get install make		# 下载GNU Make编译器

sudo apt-get install initramfs-tools   # 下载安装initramfs-tools

步骤二：禁用nouveau通用驱动

修改blacklist.conf文件

sudo vim /etc/modprobe.d/blacklist.conf

在blacklist.conf文件，最下方添加命令

blacklist nouveau
options nouveau modeset=0

执行以下命令使禁用生效并且重启：

sudo update-initramfs -u
sudo reboot

重启后可以验证是否生效：

lsmod | grep nouveau

若没有输出，则禁用生效。

步骤三： 查看GPU型号，并下载对应驱动

1.打开终端输入以下命令，获取显卡驱动信息

ubuntu-drivers devices

2.选择推荐的驱动版本进行安装

注意：显卡驱动版本和cuda版本不一样，显卡版本按推荐的配置就好。

sudo apt install nvidia-driver-535
sudo reboot #安装完成后重启
nvidia-smi #如果出现GPU列表，则驱动安装成功

2. 安装CUDA

为什么要安装11.3版本，因为11.3版本适配的pytorch版本是最多的，很多像YOLO模型，需要专门的pytorch版本，否则全是报错

在官网下载cuda，版本根据自己的选择来，依然强烈推荐11.3
[CUDA官网](https://developer.nvidia.com/cuda-toolkit-archi
按图中进行选择

输入对应的命令

wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installer/cuda_11.3.0_465.19.01_linux.run
sudo sh cuda_11.3.0_465.19.01_linux.run

选择continue

输入accept
这里有一点需要注意一下，因为之前已经安装过显卡驱动了，所以这里需要取消Driver（光标移动到Driver，然后按enter即可取消，不取消会导致安装终止），然后移动光标到Install，按enter执行安装。
接下来进行环境配置

nano ~/.bashrc      #打开bashrc脚本文件
export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc    #脚本文件生效

nvcc -V

安装成功

2. 安装CUDNN

1. 下载CUDA版本对应的cuDNN版本 cudnn官网
   可以选择11.3或者11.x
   cudnn安装包的copy

sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h 

验证

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

显示对应的cudnn 版本是8.9.2，安装成功