Windows环境下使用 LibTorch(1.4)

最近使用LibTorch在调用分类模型,在配置环境时,用到了1.4版本的PyTorch,按照网上的一些教程无法成功,便把自己部署成功的经验分享出来。

1. 从官网下载已经编译好的LibTorch库

官网地址:(https://pytorch.org/

根据自己的环境可以下载GPU版和CPU版。

2. 使用cmake命令创建项目,失败!

安装使用可参考官网文档:https://pytorch.org/cppdocs/installing.html

遇到的问题:

使用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++版本。

猜你喜欢

转载自blog.csdn.net/Challovactor/article/details/104793002
1.4