YOLO安装（Nvdia GPU）

安装所需附件下载

YOLO安装所需依赖库以及测试文件资源-CSDN文库​​​​​​​

Nvdia驱动安装

首先需要根据显卡型号安装cuda和cudnn

创建虚拟环境

方法有很多，这里使用virtualenv

安装pytorch

可以到pytorch官网查询安装命令：pytorch官网

根据cuda版本选择命令

可能会碰到官网推荐命令和本地版本不符，直接改变最后的版本号就可以了，比如图中官网推荐是118，本地版本是116，只要把命令中的118改成116就可以了

查看是否安装成功

import torch
x = torch.rand(5, 3)
print(x)

print(torch.cuda.is_available())

安装pycocotools

Pip install pycocotools-windows

安装其他依赖库

pip install -r requirements.txt

运行detect.py

python.exe .\detect.py --source .\data\images\bus.jpg --weights .\pretrained\yolov5s.pt

运行的时候有可能会报以下错误

AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'

那是因为yoloV5的版本太高了，需要更改源码，更改完成后运行成功