怎么编译pytorch

编译 PyTorch 通常需要以下步骤：

1.安装 Anaconda 和 CMake：如果没有安装，请先安装 Anaconda 和 CMake。
2.安装 PyTorch 依赖项：在命令行中输入以下命令来安装 PyTorch 依赖项：

conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six

3.获取 PyTorch 源代码：可以从 PyTorch 官网下载 PyTorch 的源代码，也可以使用以下命令从 GitHub 上获取 PyTorch 源代码：

git clone --recursive https://github.com/pytorch/pytorch

4.设置环境变量：在命令行中设置以下环境变量：

export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
export TORCH_CUDA_ARCH_LIST="7.0"

其中，`TORCH_CUDA_ARCH_LIST` 的值取决于你的 GPU 架构，可以根据自己的情况进行设置。

5.编译 PyTorch：在源代码目录中运行以下命令编译 PyTorch：

python setup.py install

这会在当前 Anaconda 环境中安装 PyTorch。

如果你需要编译 PyTorch 的 CUDA 扩展模块，可以在源代码目录中运行以下命令：

python setup.py install --user

这会在当前用户目录下安装 PyTorch。