yolov5从GPU安装到训练自己的模型详细教程

                      yolov5从GPU安装到训练自己的模型详细教程 

环境搭建
一． Gpu安装

1 、linux系统

查看自己电脑显卡型号:
linux系统终端指令：lspci | grep –i nvidia ,会返回gpu型号，如果没有返回你的电脑可能没有gpu显卡或显卡未正确安装。

禁用nouveau
使用如下命令查看系统是否正在使用nouveau：

如下图，如果有任何输出，那么就是nouveau在启用，需要按照以下步骤关闭，否则跳过禁用nouveau部分即可。

创建配置文件
sudo vi /etc/modprobe.d/blacklist-nouveau.conf
加入以下内容，保存退出
blacklist nouveau
options nouveau modeset=0
执行如下命令，更新镜像
sudo update-initramfs -u
等待更新完后，用reboot命令重启系统，再次使用lsmod | grep nouveau核实无任何输出。
cuda的安装

在NVIDIA官网找到CUDA各版本下载页面，链接https://developer.nvidia.com/cuda-toolkit-archive，如下图：

选择CUDA Toolkit 10.2，在新窗口选择操作系统版本和CPU架构等信息，安装类型选择runfile[local]，可以离线安装。（官网更推荐deb包的安装方式，但是我没成功，有兴趣可以到官网学习，资料非常详细）

终端执行：
Wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.runsudo
sh cuda_10.2.89_440.33.01_linux.run
进行下载安装cuda。

等待安装程序加载，过会儿会询问是否接受协议，输入accept回车确认

然后会询问安装的组件，全选即可，选择好之后把光标移动到Install上面，按下回车键，等待安装。

安装完成后，修改环境变量，执行sudo vi ~/.bashrc打开配置文件，在最后加入以下内容并保存退出。
export PATH=/usr/local/cuda-10.2/bin:/usr/local/cuda-10.2/nsight-compute-2019.5.0: P A T H e