Yolo GPU训练（python）

学习过程记录，看爽了你就点个赞
首先你得有显卡，且支持NVIDIA CUDA
驱动安装
​​更新显卡驱动​​：
访问 NVIDIA 驱动下载
选择对应显卡型号下载最新驱动
​​验证驱动安装​​：
命令行里输入：nvidia-smi
应该显示 GPU 信息和 CUDA 版本（如 CUDA 12.x）

软件环境安装
安装 CUDA 和 cuDNN
​​安装 CUDA Toolkit​​：
下载地址：https://developer.nvidia.com/cuda-toolkit-archive
选择 CUDA 11.8（YOLOv8 推荐版本）
安装时勾选 “Visual Studio Integration”
​​安装 cuDNN​​：
下载地址：https://developer.nvidia.com/cudnn
需要注册 NVIDIA 账号
下载与 CUDA 版本匹配的 cuDNN
解压后将 bin、include、lib 文件夹内容复制到 CUDA 安装目录(这时有相同的文件夹结构，bin -> bin,include ->include,lib->lib)别一股脑乱放。
​​设置环境变量​​：
确保以下路径已添加到系统 PATH：
C:\Program Files\NVIDIA GPU