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安装及配置_第1张图片

下载解压之后如下

Libtorch + vs 2019安装及配置_第2张图片

2、在VS2019中的配置

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

配置头文件

Libtorch + vs 2019安装及配置_第3张图片

配置依赖库的位置

Libtorch + vs 2019安装及配置_第4张图片

添加依赖库的文件名

Libtorch + vs 2019安装及配置_第5张图片

设置dll文件的环境变量,

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

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

Libtorch + vs 2019安装及配置_第6张图片

4、简单的测试例子

#include 
#include 

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

    return EXIT_SUCCESS;
}

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

Libtorch + vs 2019安装及配置_第7张图片

 

 

你可能感兴趣的:(人工智能,libtorch,c++)