注意的事项:
windows下如果已经安装过普通的驱动,CUDA不会替换已安装的驱动 - 它们是互补关系
驱动是基础 - 让显卡能正常工作
CUDA是扩展 - 让显卡能进行AI计算
版本要兼容 - CUDA版本不能超过驱动支持的最高版本!!
安装cuda驱动时:
# 当安装程序出现组件选择时:
[ ] Driver
# ↑ 取消勾选!因为你已经有更新的驱动了
[✓] CUDA
[✓] Development
[✓] Runtime
[✓] Samples
[✓] NSight
1:yolo使用gpu需要安装nvidia cuda的驱动,在安装cuda之前,需要先检查当前驱动版本:
输入nvidia-smi查看支持cuda最高的版本,显示为12.4,那么cuda的驱动版本不可以超过12.4
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 552.22 Driver Version: 552.22 CUDA Version: 12.4 |
|-----------------------------------------+------------------------+----------------------+
2:安装好cuda驱动后,nvidia会自动添加好环境变量,但windows可能需要重启使环境变量生效,否则没有nvcc的命令。
输入nvcc --version 查看cuda安装好了没有
>nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Thu_Mar_28_02:30:10_Pacific_Daylight_Time_2024
Cuda compilation tools, release 12.4, V12.4.131
Build cuda_12.4.r12.4/compiler.34097967_0
3:python这里的torch需要使用cuda的版本
下面的cu124就代表对应的cuda12.4版本
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
1. 模型架构和大小
bash
# YOLOv5不同尺寸的模型对比
yolov5n.pt (纳米) → 精度低,速度极快
yolov5s.pt (小) → 精度中等,速度快
yolov5m.pt (中) → 平衡性好
yolov5l.pt (大) → 精度高,速度慢
yolov5x.pt (超大) → 精度最高,速度最慢
