libtorch c++调用 (八)模型加载失败原因总结

编程入门 行业动态 更新时间:2024-10-12 10:22:41

libtorch c++调用 (八)<a href=https://www.elefans.com/category/jswz/34/1771358.html style=模型加载失败原因总结"/>

libtorch c++调用 (八)模型加载失败原因总结

1、最常见原因是:pytorch版本与libtorch版本不一致。虽然偶尔有使用低版本保存的模型高版本libtorch可以调用的情况,但是一般高版本的pytorch保存的模型,低版本的libtorch加载就会报错。所以一般libtorch版本不能比pytorch版本低,但是最好二者版本一致。
2、找不到运行库或运行库不全。这个一般在调试时将debug目录设置为生成可执行程序的目录并且将libtorch库拷贝到该目录下就可以调试。
3、pytorch保存的gpu模型cpu加载,加载时没有设置设备类型导致的加载失败。

比如下面这样加载失败:

torch::jit::script::Module module = torch::jit::load("./script_test.pt");

而这样加载就可以加载成功:

torch::Devic

更多推荐

libtorch c++调用 (八)模型加载失败原因总结

本文发布于:2024-02-16 18:27:52,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1691055.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:模型   加载   原因   libtorch

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!