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