Libtorch安装与使用
Libtorch安装与使用
LibTorch 是 PyTorch 的 C++ 接口，它提供了一种在 C++ 环境中使用 PyTorch 深度学习框架的方式，也就是libTorch就是一种深度学习的框架。
libTorch安装
下载网站：PyTorch
进入首页后，下拉找到自己想要安装的版本
解压后将 .../libtorch/lib添加到环境变量里面
点击编辑，进入后点击添加，将.../libtorch/lib添加即可
LibTorch使用
环境：vs2019
- 创建一个新项目，修改项目的配置
这里使用DEBUG x64
第一处修改，选择DEBUG X64
第二处修改，右键点击解决方案->点击配置管理器->选择DEBUG X64
- 添加头文件，库文件，以及指定连接的库文件
第一处修改，右键点击项目->点击属性->点击C/C++ ->点击附加包含目录->编辑
输入：
...\libtorch\include
...\libtorch\include\torch\csrc\api\include
... 是你自己的路径
PS: 如果你的属性点击后没有C/C++请检查是否安装C/C++桌面开发，在Vstudio install 检查一下，如果没有安装，安装后重启软件即可
第二处修改，右键点击项目->点击属性->点击链接器->点击常规->点击附加库目录->点击编辑->添加附加库目录：
...\libtorch\lib
第三处修改，右键点击项目->点击属性->点击链接器->点击输入->点击附加依赖项，点击编辑->输入：
里面的内容就是你需要的库目录
PS:如果你不知道你需要哪些库目录，那么你可以将lib库里面所有.lib的文件都包含进去
就可以开始使用了
测试demo：
#include <iostream>#include <torch/torch.h> // LibTorch 头文件，包含了神经网络与张量相关功能int main(){//1. 创建一个 2*3 的张量torch::Tensor tensor = torch::randn({2 * 3});std::cout << "tensor: \n" << tensor << std::endl;//2. 张量的加法tensor += 1;std::cout << "\ntensor: \n" << tensor << std::endl;return 0;}
只要有结果，那就说明头文件和库文件没什么问题