Libtorch + vs 2019安装及配置

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

1、从官网下载

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

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

Libtorch + vs 2019安装及配置

 

下载解压之后如下

Libtorch + vs 2019安装及配置

 

2、在VS2019中的配置

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

配置头文件

Libtorch + vs 2019安装及配置

 

配置依赖库的位置

Libtorch + vs 2019安装及配置

 

添加依赖库的文件名

Libtorch + vs 2019安装及配置

 

设置dll文件的环境变量,

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

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

Libtorch + vs 2019安装及配置

 

4、简单的测试例子

#include <iostream>
#include <torch/torch.h>
 
int main()
{
    
    torch::Tensor tensor = torch::rand({ 5,3 });
    std::cout << tensor << std::endl;
 
    return EXIT_SUCCESS;
}


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

Libtorch + vs 2019安装及配置

 

上一篇:libtorch教程(四)


下一篇:基于RV1126平台imx291分析 --- 基于subdev的方式