LibTorch的安装、配置与使用

LibTorch的安装、配置与使用

  • LibTorch的安装
    • LibTorch的下载与环境变量的设置
      • (1)解压压缩包
      • (2)设置环境变量:
    • VS2019的配置
      • (1)设置配置管理器:
      • (2)设置头文件路径:
      • (3)设置链接库:
    • 测试代码

LibTorch的安装

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

在这里下载: Start Locally|PyTorch.
LibTorch的安装、配置与使用_第1张图片
上图中有两个下载链接,一个是release版本一个debug版本。
有C++开发经验的小伙伴应该了解如何使用这种库:

(1)解压压缩包

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

(2)设置环境变量:

右键我的电脑->属性->高级系统设置->高级中的环境变量->点击系统变量中的Path->添加dll路径:
LibTorch的安装、配置与使用_第3张图片

VS2019的配置

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

(1)设置配置管理器:

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

(2)设置头文件路径:

LibTorch的安装、配置与使用_第6张图片
在这里进行设置。

LibTorch的安装、配置与使用_第7张图片
这两个头文件路径中常用的头文件分别是:

#include "torch/script.h"

#include "torch/torch.h"

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

(3)设置链接库:

添加LibTorch包含lib的文件夹路径
LibTorch的安装、配置与使用_第8张图片
添加所需的lib文件
LibTorch的安装、配置与使用_第9张图片
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;
}

运行得到结果:
LibTorch的安装、配置与使用_第10张图片

你可能感兴趣的:(LibTorch之旅,c++,pytorch,机器学习,神经网络,windows)