PyTorch安装c++
时间: 2023-12-21 20:06:39 浏览: 158
安装 C++ 版本的 PyTorch 需要先安装 CMake 和 C++ 编译器,推荐使用 Visual Studio 或者 Clang。
以下是在 Windows 下安装 C++ 版本的 PyTorch 的步骤:
1. 安装 Visual Studio 或者 Clang,建议使用 Visual Studio 2019。
2. 下载并安装 CMake。
3. 安装 Anaconda 或者 Miniconda。
4. 打开 Anaconda Prompt 或者 Miniconda Prompt,创建一个新的虚拟环境:
```
conda create -n torch_cpp python=3.7
```
5. 激活虚拟环境:
```
conda activate torch_cpp
```
6. 安装 PyTorch:
```
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
注意,这里使用的是 `cpuonly` 版本,如果需要使用 GPU 版本,则需要安装对应的 CUDA。
7. 使用 CMake 进行编译:
```
git clone --recursive https://github.com/pytorch/pytorch.git
cd pytorch
mkdir build
cd build
cmake .. -DCMAKE_PREFIX_PATH="$(python -c 'import torch;print(torch._C._get_install_prefix())')" -DCMAKE_CXX_FLAGS="-D_GLIBCXX_USE_CXX11_ABI=0"
cmake --build .
```
以上步骤完成后,就可以在 C++ 中使用 PyTorch 了。
阅读全文