美文网首页
LibTorch安装&配置&使用

LibTorch安装&配置&使用

作者: WAI_f | 来源:发表于2020-05-05 06:03 被阅读0次

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

    相关文章

      网友评论

          本文标题:LibTorch安装&配置&使用

          本文链接:https://www.haomeiwen.com/subject/aslpghtx.html