libtorch在windows上的使用及采坑

已于 2022-09-07 15:45:16 修改
阅读量990 收藏
点赞数
分类专栏： AI部署 文章标签： 深度学习 深度学习部署
于 2022-09-07 15:38:07 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/whunamikey/article/details/126746952
版权

1.环境搭建

2.将pytorch模型导出成pt

3.使用libtorch c++ api调用加载模型

4.errorlist#1:libtorch在windows上部署时出现这个报错，模型无法加载：

libtorch在windows上部署时出现这个报错，模型无法加载；解决方案：

在C++->属性->链接器->命令行添加：

/INCLUDE:"?warp_size@cuda@at@@YAHXZ"

/INCLUDE:"?ignore_this_library_placeholder@@YAHXZ"

首先得确保你的libtorch和torch版本一致，不然都是白搭

在使用debug下使用CPU能进行模型测试，但使用GPU出现C10：：error；

在img.to(device)->img=img.to(device)

5.errorlist#2: Relsease或者Debug只有一个能够跑通：

解决方案：debug和release需要分开配置，库不能够共用

6.参考链接

libtorch使用教程，知乎上总结

libtorch (pytorch c++) 教程（一） - 知乎

mikeyNa
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
C++下的libtorch部署——（三）window10下的Visual Studio配置libtorch
qq_43199575的博客
08-24 4634
libtorch可以理解为C++版的pytorch，语法和pytorch基本相似。安装libtorch之前，要先安装好cuda以及对应的pytorch。
libtorch_vs2017_v141_win_shared_with_deps_1.5.0.zip
05-09
libtorch的vs2017 64位 release 的C++开发包,平台集v141。include配置【libtorch\include ；libtorch\include\torch\csrc\api\include】；lib配置【libtorch\lib ； caffe2_detectron_ops.lib、caffe2_module_test_dynamic.lib、torch.lib、torch_cpu.lib、c10.lib、fbgemm.lib、cpuinfo.lib、clog.lib、libprotoc.lib、libprotobuf.lib、mkldnn.lib、libprotobuf-lite.lib、asmjit.lib】；属性->C/C++ ->常规->SDL检查->否；属性->C/C++ ->语言->符号(合)模式->否； 测试例程： #include #include struct Net : torch::nn::Module { Net(int64_t N, int64_t M) { W = register_parameter("W", torch::randn({ N, M })); b = register_parameter("b", torch::randn(M)); } torch::Tensor forward(torch::Tensor input) { return torch::addmm(b, input, W); } torch::Tensor W, b; }; int main() { std::cout << "Hello World!\n"; torch::Tensor tensor = torch::eye(3);// torch::eye(3); std::cout << tensor << std::endl; Net net(4, 5); for (const auto& p : net.parameters()) { std::cout << p << std::endl; } system("PAUSE"); }
Pytorch的C++端(libtorch)在Windows中的使用
老潘的博客
01-23 9974
关于本文章的最新更新请查看:oldpan博客 前言 填一个之前的坑啊，本篇的姊妹篇——利用Pytorch的C++前端(libtorch)读取预训练权重并进行预测 这篇文章中已经说明了如何在Ubuntu系统中使用libtorch做预测，当初也有朋友问我如何在Windows之下尝试使用libtorch，当时因为时间关系没有去看，后来就给忘了…现在有时间了当然要尝试一下~ 当然先说结论哈，其实在Wind...
libtorch在windows使用（c++项目中部署.pt）
Already8888的博客
01-10 1499
PyTorch 提供TorchScript，加载保存的策略，除了 PyTorch 之外不需要其他依赖项。这在将训练的模型部署到生产环境时特别有用。 TorchScript模型可以加载在c++中，嵌入系统项目使用。
Windows 下通过Libtorch_CPU部署 yolo v5-6.0
qq_44699829的博客
07-22 1133
基于libtorch cpu版本的yolov5部署
在clion上配置libtorch开发环境的图文详解
01-21
先上结果： CPP： CMakeLists.txt: 到这里差不多的选手应该都不需要往下看了 描述问题： 按照pytorch官网给的教程，在...因此本文说明如何在clion上配置libtorch开发环境 报错：  解决办法： 将CMakeLists.txt
libtorch安装包
03-26
**libtorch** 是一个由PyTorch团队推出的C++库，它提供了深度学习框架的核心组件，使得开发者可以在不使用Python的情况下进行高效的机器学习和深度学习应用开发。这个库包含了Tensor运算、自动求导机制和CUDA支持，...
Jetson TX2下编译AArch64版本的libtorch
最新发布
02-21
在深入探讨Jetson TX2上编译AArch64版本的libtorch之前，我们先了解一下基本概念。libtorch是PyTorch的一个轻量级库，它提供了C++接口，允许开发者在C++环境中构建和运行深度学习模型。PyTorch是一个广泛使用的开源...
libtorch.zip
08-24
这个32位版本特别针对Windows操作系统，这意味着它可以在运行32位软件的系统上使用。然而，值得注意的是，由于是CPU版本，它并未集成CUDA支持，因此无法利用GPU加速计算，这对于依赖高性能计算的大型模型可能会有所...
PyTorch-CPP：使用LibTorch进行PyTorch C ++推理
01-28
该演示将演示如何使用LibTorch来构建C ++应用程序。 [UPDATE 2019/01/18] ：初始化仓库，使用PyTorch1.0进行测试。 [UPDATE 2020/02/22] ：感谢和他的 ，他们更新了此演示以适合LibTorch1.4.0和OpenCV4.0。 [UPDATE...
Libtorch-Windows 配置使用
桃子好好吃的博客
10-24 717
使用c++调用pytorch模型时，如果使用Libtorch会更加简洁（不需要额外的python文件辅助）。此博客记载Libtorch的配置过程。使用系统及工具（Windows10 & VS 2019）. 下载与当前安装的pytorch版本相同的libtorch，比如，我的电脑安装的pytorch是v1.6，libtorch也应该下载v1.6。 解压下载好的压缩包，保存在自己的目录下 VS 2019 配置 ...
Windows下安装libtorch与Clion配置
weixin_42216799的博客
12-16 1315
windows下为Clion配置libtorch和使用
windows10下c++加载libtorch
weixin_42425256的博客
07-25 725
windows10下c++加载libtorch
C++下的libtorch部署——（二）window10下的Visual Studio配置OpenCV
qq_43199575的博客
08-23 557
在进行图像处理时基本上都会用到OpenCV，所以需要将其配置到Visual Studio当中。直接到官网选择自己需要的版本下载即可，下载解压完会有一个“opencv”文件夹，其中的“bulid”文件夹就是需要配置的关键路径。下载地址。
LibTorch中Windows系统环境配置及CUDA不可用问题解决
weixin_42112050的博客
03-24 2152
在 Python 深度学习圈，PyTorch 具有举足轻重的地位。同样的，C++ 平台上的 LibTorch 作为 PyTorch 的纯 C++ 接口，它遵循 PyTorch 的设计和架构，旨在支持高性能、低延迟的 C++ 深度学习应用研究。本文基于 Windows 环境与 Visual Studio 2022 开发工具，将从零开始搭建一个完整的深度学习LibTorch开发环境。
windows下libtorch配置指南
HKTXT的专栏
04-23 732
libtorch配置指南 1. 安装cuda toolkit https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal 下载自己对应的版本安装 2. 安装CuDNN htt...
Pytorch1.0的（C++ & cpu）版本在Windows上的使用
weixinhum
03-15 2374
之前一直用Pytorch做的网络训练，最近想看看训练出来的模型在C++上的表现（毕竟稍微注重效率的工程都用的C++写），因此花了点时间研究了一波。 因为PyTorch的Python版本和C++版本使用的文件描述有区别，因此需要先把Python版本训练的模型转换成Torch Script形式（相信这一点不太方便的地方后面会被FB大佬们改掉的）。 转换的代码如下： import torch impor...
Windows使用C++调用Pytorch1.0模型
TeddyZhang的博客
12-20 5143
最近因为项目需求，而且面对的是一些用户级的，也就是windows客户端，所以就在思考在windows下如何调用深度学习模型，做了以下尝试： 使用Caffe框架，虽然也可以，但我的大部分项目都使用的Pytorch，真的不想去碰Caffe，而且修改不能按照自己的思想，pass! 使用ONNX，当时由于Pytorch1.0并没有发布windows版，我们想着把Pytorch模型转换为ONNX，然...
libtorch使用记录
weixin_41054702的博客
06-23 270
1.pytorch -> libtorch 模型转换脚本 num_classes = 3 model = DeepLabV3Plus(nclass = num_classes) checkpoint = torch.load("/data_1/project/nanchang/seg/deeplabv3+torch/model-800/cur_itrs_99000_os8.pth") model.load_state_dict(checkpoint['model_state
Windows系统安装libtorch库
05-18
安装 libtorch 库需要以下步骤： ... 2. 点击 “Get Started” -> “Quick ...以上就是安装 libtorch 库的步骤，需要注意的是，不同版本和配置的 libtorch 库在安装和使用过程中可能存在差异，需要根据实际情况进行调整。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值