libtorch部署2:用VC加载模型进行推理

前面已经生成了使用于libtorch的模型,下面就在VC2015下进行部署。

基本环境
操作系统:Win10
编译器:VS 2015
Cuda版本:CUDA10+cuDNN7.6.5
Python版本:Python 3.6.1
Pytorch版本:1.4.0
Libtorch版本:1.4.0

1.下载libtorch

如果你用的最新的,那就从官方地址下载就好了:
https://pytorch.org/get-started/locally/

我这儿是采用的1.4.0版本,下载地址如下:
https://download.pytorch.org/libtorch/cu101/libtorch-win-shared-with-deps-1.4.0.zip

下载后解压可以得到:
在这里插入图片描述

2.新建VC工程

头文件,就直接拷贝 include目录就可以了。

include

库文件,如果用cuda就用带cuda后缀的库

c10.lib或c10_cuda.lib
libprotobuf.lib
torch.lib 或 torch_cpu.lib 或 torch_gpu.lib

关键点来了,在程序中引用的时候要加上:

#undef UNICODE   //这个一定要加,否则会编译错误
#include <torch/script.h> // 引入libtorch头文件

3.代码示例

测试中需要用opencv库来处理图片,直接在官网获取即可。
https://opencv.org/releases/
解压后得到对应的头文件和库,这个就不赘述了。

找一张手写字体图来进行测试

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值