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、简单的测试例子
#include <iostream>
#include <torch/torch.h>
int main()
{
torch::Tensor tensor = torch::rand({ 5,3 });
std::cout << tensor << std::endl;
return EXIT_SUCCESS;
}
运行结果如下,说明配置没问题,可以开始后面的Libtorch学习之旅。