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;
}
运行得到结果: