订阅专栏
最近想研究一下如何部署pytorch模型,看到别人说LibTorch不错,是最近流行的C++部署框架,我打算试试。
首先,去pytorch官网上下载,如下图,上面的连接是release版本,下面是Debug版本,支持C++,但是Window不支持Java。点击连接下载下来。我选择的是release版本的。
将解压后的文件复制到C盘,如下图:
设置环境变量:
右键我的电脑->高级系统设置->高级中的环境变量->点击系统变量中的Path->添加dll路径:
.
然后新建C++工程文件
新建完成后默认是x86的,需要改为X64的,因为libtorch是x64的,不支持32位的。我下载是release版本的,所以还要改为release模式。
然后右键lib_torch项目,选择属性,配置libtorch环境。
选择C\C++ ——》常规,将SDL检查改为否。添加附加包含目录。
改完后的效果如下图:
将 C/C++ => 语言 => 符合模式,改为否。
设置连接库,选择连接器——》常规,修改附加库目录
设置连接库,选择连接器——》输入,修改附加依赖项
在附件依赖项中增加:
c10.lib
mkldnn.lib
libprotobuf.lib
torch.lib
torch_cpu.lib
在 配置属性à调试à环境 里添加libtorch动态库的路径
PATH=C:\libtorch\lib;%PATH%
到这里就配置完了,写入测试代码试一下。
#include <torch/torch.h>
#include "torch/script.h"
#include <iostream>
int main()
{
torch::Tensor output = torch::randn({ 5,2 });
std::cout << output;
return 0;
}
出现下面的结果说明配置成功了。