c++安装PyTorch
时间: 2023-12-21 09:06:42 浏览: 35
要在 C++ 中使用 PyTorch,您需要先安装 PyTorch C++ 库。以下是在 Windows 和 Linux 上安装 PyTorch C++ 库的步骤:
### Windows
1. 在 https://pytorch.org/ 网站上下载适用于 Windows 的 PyTorch 安装程序。
2. 运行安装程序并选择 “Custom” 安装选项。
3. 在选择要安装的组件时,选择 “C++” 组件并继续安装。
4. 安装完成后,在 Visual Studio 中打开您的 C++ 项目。
5. 在项目属性中,将 “VC++ Directories -> Include Directories” 添加到 PyTorch C++ 库头文件的路径。
6. 将 “VC++ Directories -> Library Directories” 添加到 PyTorch C++ 库文件的路径。
7. 在 “Linker -> Input -> Additional Dependencies” 中添加以下库文件:
`torch.lib;caffe2.lib;c10.lib`
### Linux
1. 在 https://pytorch.org/ 网站上根据您的 Linux 发行版选择适当的安装方式。
2. 使用以下命令安装 PyTorch C++ 库:
```
pip install torch==1.8.0+cpu -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
如果您使用 GPU 版本,请将 `+cpu` 替换为 `+cu111`。
3. 在您的 C++ 项目中,将 PyTorch C++ 库头文件的路径添加到包含目录中。
4. 将 PyTorch C++ 库文件的路径添加到链接器库路径中。
5. 在链接器选项中添加以下库文件:
`-ltorch;-lcaffe2;-lc10`