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

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

他人的解决方案

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

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

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

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

我的解决方案

在查阅很久没有找到合适的解决方案的时候,我想着是不是自己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. 最后复制该命令到终点,进行安装对应版本的库。安装成功后如下:
    在这里插入图片描述
  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

心无旁骛~

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值