yolov5-调用GPU

最新推荐文章于 2026-04-09 09:07:53 发布
原创 最新推荐文章于 2026-04-09 09:07:53 发布 · 1.5w 阅读 · 56 ·
本内容遵循CC 4.0 BY-SA版权协议
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
收录于
pytorch

当前文章被收录于：

pytorch
148 篇文章 31 人学习
订阅专栏 查看详情
本文详细阐述了在调试Yolov5时遇到的CPU设备提示问题，提供了解决方案，即在detect.py文件中将device参数的默认值从'CPU'改为''，确保正确配置GPU使用。

文章目录

1.遇到问题

在调试yolov5时，我们发现在运行detect.py时发现提示是：using CPU

2.解决方式

  • 将device中的default =“CPU”改成默认“ ”；注意：在模块：if name == ‘main’:中
    更改前：
parser.add_argument('--device', default='CPU', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')

更改后：

parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
YOLOv5-6.2源码解析-train.py(超级无敌巨详细版)
m0_74025441的博客
06-05 3741
本人炼丹小白，此篇博文用于记录学习YOLO系列之旅，一方面供自己回顾，一方面供大家参考，欢迎各路大神评论交流。所写内容均为原创，如需参考引用请标明出处！
yolov5-6.x训练参数
Scream_1的博客
04-18 4840
yolov5训练参数设置
YOLOv5-推理时使用GPU
weixin_39357271的博客
03-04 1万+
YOLO-v5，GPU
yolov5检测代码detect.py参数注释及解析
weixin_39128932的博客
01-30 7212
参考了大佬的文章https://blog.csdn.net/Q1u1NG/article/details/108093525 参数及main函数解析 if __name__ == '__main__': """ weights:训练的权重 source:测试数据，可以是图片/视频路径，也可以是'0'(电脑自带摄像头),也可以是rtsp等视频流 output:网络预测之后的图片/视频的保存路径 img-size:网络输入图片大小 conf-thres:置信度阈值
关于python中parser.add_argument的用法
m0_56514535的博客
06-28 1627
name or flags - 选项字符串的名字或者列表，例如 foo 或者 -f, --foo。argparse 模块是 Python 内置的一个用于命令项选项与参数解析的模块。
一行代码解决yolov5GPU调用问题
sp7414的博客
04-03 8358
利用yolov5做项目期间，由于要同时训练两个项目代码，发现一个之前没有发现的问题：GPU调用默认从0开始。 举个例子：train.py中device调用代码 parser.add_argument('--device', default='4,5,6,7', help='device id (i.e. 0 or 0,1 or cpu)') 比如我这里在这里调用 device = 4,5,6,7 。按理来讲，实验室服务器8块GPU，应该调用后面四个才对。然而实际上它默认从0开始，调用了0,1,2,3四块G
yolov5踩过的初坑（关于detect代码可以调用gpu而训练代码无法调用gpu的问题）
y1358898035的博客
09-24 9688
第一次写博客，如有不妥之处，还请谅解。 初遇这个问题是在笔记本上测试训练代码遇到的，起初以为是笔记本配置太低（windows10 64位 显卡NVIDIA GeForce MX350, 2048.0MB），后来又用服务器试了试，依旧是存在detect可调用gpu，但训练代码无法调用gpu的问题，检查了两天发现问题，并已解决。 yolov5原理+环境搭建+代码测试啃了一个月，还不是很熟，环境搭建途中遇到的问题也是各种找度娘，各种看博客，发现有个问题，我遇到了，但是百度不到，所以就把我的解决办法发出来，如果
YoloV5程序打包成exe文件，无法调用GPU
weixin_46544694的博客
06-11 342
摘要：用户提问打包后的exe文件无法调用GPU，怀疑与未安装实际环境下的CUDA有关。原py文件在虚拟环境中可正常使用CUDA，但打包后失去GPU调用能力。寻求解决方案，希望了解是否因缺少真实环境的CUDA驱动导致该问题。（49字）
在Windows环境下调用GPU部署yolov5模型
weixin_43193894的博客
08-16 1854
在win10+opencv4.4.0+opencv_contrib4.4.0+cuda10.2环境下部署yolov5模型 编译过程参考了这一篇博客 https://blog.csdn.net/hai_fellow_Z/article/details/116002203 主要有两个代码yolo.h和main_yolo.cpp 主程序为main_yolo.cpp文件 以下代码main函数可以分别实现图片、视频、摄像头的调用 #include "yolo.h" #include <opencv2/op
YOLOv5免环境训练工具 懒人精灵yolov5免环境训练工具 使用懒人精灵官方代码即可直接调用，小白也能轻松上手，内部封装cuda114使用GPU训练
最新发布
2401_88440984的博客
04-09 236
YOLOv5免环境训练工具 懒人精灵yolov5免环境训练工具 使用懒人精灵官方代码即可直接调用，小白也能轻松上手，内部封装cuda114使用GPU训练
基于OpenCV GPU的YOLOv5图片检测
深之JohnChen的专栏
09-26 865
3）、调用cv::dnn::Net类中setPreferableBackend和setPreferableTarget方法。2）、cv::dnn::readNetFromONNX加载yolov5 onnx模型文件。1）、cv::cuda::setDevice设置使用gpu id。1、编译好OpenCV GPU版本库。2、调用DNN模块方法。
【学习总结】win10下yolov5使用GPU训练自己的模型，同时调用网络摄像头识别目标、学习指南与踩坑经验
热门推荐
m0_55317949的博客
08-24 3万+
yolov5学习经验，纯新手向，包学包会
windows下yolov5使用GPU训练自己的模型，同时调用网络摄像头识别
m0_62024035的博客
11-10 618
5、开始训练，运行mytrain.py。1.下载yolov5源代码存放到桌面。2.下载anaconda用于配置环境。1.修改coco128.yaml。2.修改yolov5s.yaml。3、识别网络摄像头或者流媒体。3.熟悉命令窗口简单指令。3、测试环境是否正确配置。4、修改train.py。2、识别本地图片或者视频。
【YOLOv5】LabVIEW+YOLOv5快速实现实时物体识别（Object Detection）含源码
virobotics的博客
05-24 6796
在LabVIEW中调用YOLOv5快速实现实时物体识别,感受丝滑般物体识别
懒人精灵YOLOv5免配置训练工具｜一键导出调用示例｜GPU加速（CUDA 11.4内建）｜小白友好｜支持全系列模型与ONNX/NCNN转换
htzss1990的博客
03-03 203
无需安装Python/CUDA，解压即用！内置CUDA 11.4 GPU加速，支持YOLOv5n/s/m/l/x全型号训练+模型转换，附视频教程+极速退款保障。
yolov5 c++ onnx pytorch pycharm gpu train test
u013927877的博客
02-05 1329
c++代码调用yolov5训练模型时onnx格式是一种常用格式，但是在yolov5训练模型时给出的是pt格式，因此需要将yolov5训练的 best.pt转换成best*.onnx,以便opencv c++调用。std::vector<std::string> classes这是目标分类的名称，要与训练模型时一致。（1）yolov5 pt模型转onnx模型，并用opencv c++调用；(5)模型文件best.onnx及视频文件放到当前目录，也可自行修改；方法：c++调用yolov5模型；
Windows下 YOLOv5配置、部署、及遇到的问题
Eurus_X的博客
08-20 1304
Windows下yolov5环境搭建部署及问题解决
labview yolov5 opencv dnn推理，封装dll, labview调用dll
NYTfewcsrZ的博客
01-27 830
labview yolov5 opencv dnn推理，封装dll, labview调用dll，支持同时加载多个模型并行推理，可cpu/gpu, 可识别视频和图片，只需要替换模型的onnx和nameclass即可。labview yolov5 opencv dnn推理，封装dll, labview调用dll，支持同时加载多个模型并行推理，可cpu/gpu, 可识别视频和图片，只需要替换模型的onnx和nameclass即可。这个技巧在需要频繁迭代模型的场景特别管用，比如基于在线学习的缺陷检测系统。
windows10+vs2019+opencv4.5.4实现yolov5的c++推理（gpu加速）
playezio的博客
03-22 1万+
windows10+vs2019+opencv4.5.4实现yolov5的c++推理（gpu加速）
YOLOv5模型环境搭建及使用google colab训练
kuokay的博客
04-20 4039
环境搭建 环境 ubuntu 18.04 64bit GTX 1070Ti anaconda with python 3.8 pytorch 1.7.1 cuda 10.1 yolov5 5.0.9 为了方便使用 yolov5 目标检测，有网友已经将其做成了库，提交到了官方的索引库 pypi 上，这样，我们就可以直接使用 pip进行安装了，其项目地址: https://github.com/fcakyon/yolov5-pip 安装 首先创建一个干净的 python 虚拟环境 conda create
评论 30
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
表情包 插入表情
表情包 代码片
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值