最近使用LibTorch在调用分类模型,在配置环境时,用到了1.4版本的PyTorch,按照网上的一些教程无法成功,便把自己部署成功的经验分享出来。
1. 从官网下载已经编译好的LibTorch库
官网地址:(https://pytorch.org/)
根据自己的环境可以下载GPU版和CPU版。
2. 使用cmake命令创建项目,失败!
安装使用可参考官网文档:https://pytorch.org/cppdocs/installing.html
遇到的问题:
扫描二维码关注公众号,回复:
12433653 查看本文章

使用cmake命令,并不能一次成功,主要是一些配置项不成功,如下
3. 使用CMake-GUI配置项目,成功。
在使用cmake命令不成功的情况下,可以采用cmake-gui,方便配置环境。
报 CMAKE_PREFIX_PATH未设定的错误,使用Add Entry添加该条目,并设置libtorch的路径(下载后的路径)
添加Entry后,点击Configure,配置成功,然后点击Generate生成项目。
4. 直接使用Visual Studio 创建项目,成功
1) 新建C++空项目, 选择x64,Release(根据从官网下载的不同版本确定)
2) 添加新建项目,创建main.cpp文件,添加如下代码:
#include <torch/torch.h>
#include <iostream>
int main() {
torch::Tensor tensor = torch::rand({ 2, 3 });
std::cout << tensor << std::endl;
}
3) 添加包含目录和库目录
链接器-> 输入-> 附加依赖项,添加:c10.lib; torch.lib
4) 修改C++ -> 语言 -> 符合模式,为 否
5) 编译生成,成功。
6) 运行之前需要把在torch.lib目录下的所有dll拷贝到exe目录下。
注意:
使用VS2015编译报错,错误C2872 “std”: 不明确的符号。
使用vs2017编译后,解决,猜测是libtorch1.4版本需要更高的C++版本。