(解决)C++libtorch调用torch模型(torchscript)xx处(位于 xx.exe 中)有未经处理的异常: Microsoft C++ 异常: c10::Error位于内存位置 xx

这两个异常在我这里可以归为一类

原因:python的 torch版本和libtorch 版本不同

 torch版本是cpu  而libtorch版本是cpu也不行

如果上述都没问题,那就是模型的问题,模型没有转化成功

解决:版本不同就下载重新配置训练,pytorch官网可以下载torch和libtorch

模型问题解决: 查看你libtorch的是cpu还是gpu 如果是gpu有model.to(torch::kCUDA);模型读                            取方式,那么你的静态模型也需要对应,

    打开yolov5,7.0的export.py 文件修改  device 如果你的libtorch是model.to(torch::kCUDA);

就需要改为0 反之改为cpu,这个也需要你代码里是否读取的是gpu的静态模型

转化后是这样的和yolov5 5.0不同,但是都可以读取

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

该醒醒了~

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

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

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

打赏作者

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

抵扣说明:

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

余额充值