LibTorch怎么安装
时间: 2024-06-08 07:09:33 浏览: 178
LibTorch 是 PyTorch 的 C++ API,可以用于在 C++ 中使用 PyTorch。以下是在 Linux 和 Windows 上安装 LibTorch 的步骤:
### Linux
1. 在 [PyTorch 官网](https://pytorch.org/) 下载所需版本的 LibTorch,下载链接可在页面中找到,注意选择对应的操作系统、CUDA 版本和 C++ 编译器版本。
2. 解压下载的压缩包。
3. 设置环境变量 `TORCH_LIBRARIES` 和 `TORCH_INCLUDE_DIRS`,分别指向解压后的 `lib` 和 `include` 目录。例如:
```
export TORCH_LIBRARIES=/path/to/libtorch/lib/
export TORCH_INCLUDE_DIRS=/path/to/libtorch/include/
```
4. 在 C++ 项目中添加以下编译选项:
```
-D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14 -I${TORCH_INCLUDE_DIRS} -L${TORCH_LIBRARIES} -ltorch -lc10 -Wl,-rpath=${TORCH_LIBRARIES}
```
注意将 `${TORCH_INCLUDE_DIRS}` 和 `${TORCH_LIBRARIES}` 替换成实际的路径。
### Windows
1. 在 [PyTorch 官网](https://pytorch.org/) 下载所需版本的 LibTorch,下载链接可在页面中找到,注意选择对应的操作系统、CUDA 版本和 Visual Studio 版本。
2. 解压下载的压缩包。
3. 在 Visual Studio 中创建 C++ 项目,将解压后的 `include` 目录添加到项目的包含目录中。
4. 在项目属性中的 VC++ 目录 -> 库目录中添加解压后的 `lib` 目录。
5. 在项目属性中的 C/C++ -> 代码生成 -> 运行库中选择 `多线程 DLL (/MD)`。
6. 在项目属性中的链接器 -> 输入 -> 附加依赖项中添加以下库文件:
```
torch.lib; c10.lib
```
注意将库文件路径修改为实际的路径。
7. 将解压后的 `lib` 目录添加到系统环境变量 `PATH` 中,以便程序运行时能够找到相关的库文件。
以上是在 Linux 和 Windows 上安装 LibTorch 的大致步骤,具体细节可能会有所不同。如果遇到问题,可以参考[官方文档](https://pytorch.org/cppdocs/installing.html)或在社区中寻求帮助。
阅读全文