(Yolov5常见bug)RuntimeError: “slow_conv2d_cpu“ not implemented for ‘Half‘问题解决

news/2024/12/26 21:34:17/

今天用Yolov5目标检测模型进行Flask部署的时候,遇见一个问题:RuntimeError: "slow_conv2d_cpu" not implemented for 'Half'。在王宏网上查阅了很久,发现大多数都是在讲有关Half的精度问题。
具体报错如下:
在这里插入图片描述

他人的解决方案

这里先讲下Half()函数的作用:

torch.half()函数将模型的所有浮点参数和缓冲区转换为半精度(16位)浮点数。这可以减少内存占用并提高计算效率,同时保持较高的计算精度 。

这里给出我在网上找到的一些解决方案:

  • 解决报错:“slow_conv2d_cpu“ not implemented for ‘Half‘ | 训练DreamBooth
  • 复刻yolo系列时出现的BUG及解决方法

以上问题在进行网页前端部署的时候可能解决不了问题,但是训练的时候应该还是有用的。

我的解决方案

在查阅很久没有找到合适的解决方案的时候,我想着是不是自己torch版本torchvision版本不匹配的缘故,所以我查看了自己安装的库,如下:
在这里插入图片描述
可以清晰的看到,我的torchvision版本确实和我的torch版本不太匹配,因此我进入到pytorch官网,重新安装了版本以后,我的库版本具体如下:
在这里插入图片描述
最后,完美的解决了问题,并部署成功。
在这里插入图片描述

具体解决步骤为如下:

  1. 首先,进入pytorch官网;
    在这里插入图片描述
  2. 查找到自己适应的版本,这里我们选择查看以前的版本。
    在这里插入图片描述
  3. 选择自己的版本,这里我的是torch==1.13.1+cu117的,具体命令为:pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
    在这里插入图片描述
  4. 最后复制该命令到终点,进行安装对应版本的库。安装成功后如下:
    在这里插入图片描述

http://www.ppmy.cn/news/1007405.html

相关文章

单片机外部晶振故障后自动切换内部晶振——以STM32为例

单片机外部晶振故障后自动切换内部晶振——以STM32为例 作者日期版本说明Dog Tao2023.08.02V1.0发布初始版本 文章目录 单片机外部晶振故障后自动切换内部晶振——以STM32为例背景外部晶振与内部振荡器STM32F103时钟系统STM32F407时钟系统 代码实现系统时钟设置流程时钟源检测…

【网络基础实战之路】设计网络划分的实战详解

系列文章传送门: 【网络基础实战之路】设计网络划分的实战详解 【网络基础实战之路】一文弄懂TCP的三次握手与四次断开 【网络基础实战之路】基于MGRE多点协议的实战详解 【网络基础实战之路】基于OSPF协议建立两个MGRE网络的实验详解 PS:本要求基于…

PSO粒子群优化算法

PSO粒子群优化算法 算法思想matlab代码python代码 算法思想 粒子群算法(Particle Swarm Optimization) 优点: 1)原理比较简单,实现容易,参数少。 缺点: 1)易早熟收敛至局部最优、迭代后期收敛速度慢的…

React 18 响应事件

参考文章 响应事件 使用 React 可以在 JSX 中添加 事件处理函数。其中事件处理函数为自定义函数,它将在响应交互(如点击、悬停、表单输入框获得焦点等)时触发。 添加事件处理函数 如需添加一个事件处理函数,需要先定义一个函数…

9 多进程编程

文章目录 进程以及状态进程进程的状态linux下的进程管理pstopbgkillcrontab 进程的创建两个while同时运行示例获取进程pidProcess 结构给子进程指定的函数传递参数进程间是否共享全局变量补充 进程间通信-Queue常用操作Queue 实例 管道通信(了解)样例 进…

从零开始:手把手搭建 RocketMQ 单节点、集群节点实例

🔭 嗨,您好 👋 我是 vnjohn,在互联网企业担任 Java 开发,CSDN 优质创作者 📖 推荐专栏:Spring、MySQL、Nacos、Java,后续其他专栏会持续优化更新迭代 🌲文章所在专栏&…

接口返回 blob 格式怎么转成 json

问题 工作中遇到一个接口逻辑是有时候会返回文件有时候会返回json,为了保证能接收到文件,需要将 responseType 设置为 blob。但是如果此时需要根据返回的是文件还是json来进行后续不同的操作,你在控制台输出返回的结果,会发现输出…

cURL 发送 POST 请求时收到 “Forbidden 403“ 错误

如果你在使用 cURL 发送 POST 请求时收到 "Forbidden 403" 错误,这可能是由于 CSRF(Cross-Site Request Forgery)保护功能导致的。Django 默认启用了 CSRF 保护,以防止跨站请求伪造。 为了解决此问题,你可以…