LibTorch安装&配置&使用

2020-05-05  本文已影响0人  WAI_f

LibTorch这个东西最近才了解到,试了一下,感觉用起来比想象中简单很多,因为比较喜欢用C++,所以对LibTorch很有好感,不过相关的资料还是很少,虽然网上博客都有提到如何安装,但是关于配置和使用还是资料比较少。

  1. 安装LibTorch

    上图中有两个下载链接,一个是release版本一个debug版本,有C++开发经验的小伙伴应该了解如何使用这种库:

    (1)解压压缩包: 我已经把解压文件移到我的D盘并修改了名称,这里面include是配置libtorch所需的头文件,在vs2019中配置环境的时候要添加进去;lib中有.lib和.dll两种文件,安装过opencv的小伙伴都知道,这两种文件一般是分到两个文件夹中的,.lib路径要在vs中进行设置,.dll一般要添加到环境变量中。
    (2)设置环境变量:
    右键我的电脑->高级系统设置->高级中的环境变量->点击系统变量中的Path->添加dll路径:
  2. 使用VS2019创建一个空项目:
    之所以使用VS2019主要原因是低版本的VS对C++新标准支持的程度可能比较低,在使用新版本库的时候可能会出现很多语法错误,这个是需要特别注意的。

    (1)设置配置管理器: 我安装的LibTorch是Debug版本的,所以选择Debug模式;Pytorch官网提供的库都是X64的,所以还要设置编译平台为X64。
    (2)设置头文件路径:

    我的工程里还用了opencv,所以在附加包含目录里还有opencv头文件路径,如果只是用LibTorch的话添加前两行就可以了。特别需要说明的是,这两个头文件路径中常用的头文件分别是:

#include "torch/script.h"

#include "torch/torch.h"

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

添加LibTorch包含lib的文件夹路径 添加所需的lib文件
  1. LibTorch简单C++示例
#include "torch/torch.h"
#include "torch/script.h"

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

    return 0;
}
运行得到结果:
上一篇下一篇

猜你喜欢

热点阅读