LibTorch的安装、配置与使用

LibTorch的安装

LibTorch的下载与环境变量的设置

在这里下载: Start Locally|PyTorch.
如图所示
上图中有两个下载链接,一个是release版本一个debug版本。
有C++开发经验的小伙伴应该了解如何使用这种库:

(1)解压压缩包

如图所示
我已经把解压文件移到我的D盘并修改了名称,其中include文件夹是配置libtorch所需的头文件,在vs2019中配置环境的时候要添加进去;
lib文件夹中有.lib与.dll两种文件,这两种文件一般是分到两个文件夹中的。.lib路径要在vs中进行设置,.dll一般要添加到环境变量中。

(2)设置环境变量:

右键我的电脑->属性->高级系统设置->高级中的环境变量->点击系统变量中的Path->添加dll路径:
如图所示

VS2019的配置

之所以使用VS2019主要原因是低版本的VS对C++新标准支持的程度可能比较低,在使用新版本库的时候可能会出现很多语法错误,这个是需要特别注意的。

(1)设置配置管理器:

如图所示
在这里打开配置管理器。
如图所示
根据自己安装的LibTorch版本选择选择Debug或Release模式;
另外Pytorch官网提供的库都是X64的,所以还要设置编译平台为X64。

(2)设置头文件路径:

如图所示
在这里进行设置。

如图所示
这两个头文件路径中常用的头文件分别是:

#include "torch/script.h"

#include "torch/torch.h"

网上很多的示例代码添加的是第二个头文件,但是一般都没有说这个头文件所在路径,导致程序找不到很多定义,这个问题网上提到的很少,所以在这里特别说明一下。

(3)设置链接库:

添加LibTorch包含lib的文件夹路径
如图所示
添加所需的lib文件
如图所示
lib文件如下所示

c10.lib
libprotobuf.lib
mkldnn.lib
torch.lib
torch_cpu.lib

测试代码

#include "torch/torch.h"
#include "torch/script.h"

int main()
{
    torch::Tensor output = torch::randn({ 3,2 });
    std::cout << output;

    return 0;
}

运行得到结果:
如图所示

评论 15
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值