Libtorch + vs 2019安装及配置

Libtorch是c++接口的Pytorch,提供了与python版本基本一致的接口,使用很方便,而且更适合实际部署。这里介绍Libtorch的在VS2019下的部署和简单测试使用。

1、从官网下载

https://pytorch.org/get-started/locally/

目前可以下载到Libtorch1.5版本,如果电脑支持Gpu,且安装了Cuda,可以下载Gpu版本,否则下载cpu版本。

下载解压之后如下

2、在VS2019中的配置

需要配置.头文件h 库文件,lib, 和dll文件

配置头文件

配置依赖库的位置

添加依赖库的文件名

设置dll文件的环境变量,

PATH=D:\software\gnsoftware\pyTorch\libtorch-win-shared-with-deps-debug-1.5.0\libtorch\lib;%PATH%

(之前我一般是把那一堆dll文件拷贝到生成的exe文件所在目录,导致这些dll文件会复制很多份,占据很大空间,这里通过设置依赖于项目的环境变量,少了大量拷贝,还不影响其他变量)

4、简单的测试例子

1
2
3
4
5
6
7
8
9
10
11
#include <iostream>
#include <torch/torch.h>

int main()
{
   
    torch::Tensor tensor = torch::rand({ 5,3 });
    std::cout << tensor << std::endl;

    return EXIT_SUCCESS;
}

运行结果如下,说明配置没问题,可以开始后面的Libtorch学习之旅。