C++ torch::jit::load 导入 pytorch pt模型【异常: c10::Error】解决方法

折腾了一下午(擦眼泪),留个记录吧。

debug这条路细心、耐心缺一不可(洒完鸡汤,逃)

 

事件还原:

pytorch 用 Tracing 方法导出 pt 模型后在 vs 中准备使用【torch::jit::load】deserialize模型以作预测

顺利生成解决方案,但运行时一直报错:

可能原因:

libtorch版本【libtorch-gpu-cu100-1.2.0】

pytorch版本【torch-cu101-1.6.0】【torchvision-cu101-0.8.0】

对,没错,又是该死的版本兼容!!!(土拨鼠叫.gif)

解决方法:

torch & torchvision 历史whl版本下载

从上面链接中下载与 libtorch 同版本的 torch 和 对应的 torchvision

笔者下载的是【torch-1.2.0-cp36-cp36m-win_amd64.whl】和【torchvision-0.4.0-cp36-cp36m-win_amd64.whl】

(注:cp36 指 python 3.6)

重新 pip install 即可完美解决(一群土拨鼠叫.gif)

  • 3
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 11
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值