C++:Libtorch在win10系统visual studio 2019中的配置方法

1、在https://pytorch.org中下载对应版本的libtorch。
在这里插入图片描述
2、解压后于环境变量的path中添加xxx\libtorch\lib。

3、打开vs2019建立新项目,选择release和x64模式,属性管理器中在release|x64中添加新项目属性表起名libtorch(随便起一个)。

4、包含目录添加:
xxx\libtorch\include
xxx\libtorch\include\torch\csrc\api\include

5、库目录添加:
xxx\libtorch\lib

6、链接器输入添加:
asmjit.lib
c10.lib
caffe2_detectron_ops.lib
caffe2_module_test_dynamic.lib
clog.lib
cpuinfo.lib
dnnl.lib
fbgemm.lib
libprotobuf.lib
libprotobuf-lite.lib
libprotoc.lib
mkldnn.lib
torch.lib
torch_cpu.lib
(即xxx\libtorch\lib中的所有lib文件,不同版本的libtorch不一样,可能不全需要,保险起见可以全加上)
**配置opencv的方法类似

7、相关头文件为:
#include <torch/torch.h>
#include <torch/script.h>

8、可使用以下代码测试libtorch是否正常导入:
torch::Tensor tensor = torch::rand({2, 3});
std::cout << tensor << std::endl;

**若运行出现问题,将xxx\libtorch\lib中的所有dll文件放到vs工程的exe同级目录中再尝试调试。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值