---
title: 源码编译构建
description: "获取PyTorch源码，从源码构建CUDA版本wheel并安装。"
url: https://www.hikunpeng.com/document/detail/zh/kunpengai/ecosystemEnable/pytorch/kunpengpytorch_16_0005.html
sourcePath: /source/zh/kunpengai/ecosystemEnable/pytorch/kunpengpytorch_16_0005.html
indexId: 3c8d596b7331a56ad1d6535743286c304402e59c0d5a894a99ec26c4eb6743b572
---
# 源码编译构建

获取PyTorch源码，从源码构建CUDA版本wheel并安装。

1. 获取PyTorch 2.11.0源码。
  1 2 3 git clone --recursive --branch v2.11.0 --depth 1 https://github.com/pytorch/pytorch.git cd pytorch git submodule update --init --recursive

2. 安装PyTorch构建依赖。
  1 python3 -m pip install -r requirements.txt

3. 从源码构建CUDA版本wheel。
  1 2 3 4 5 6 7 8 9 10 11 12 13 14 export CUDA_HOME=/usr/local/cuda-13.0 export PATH=/usr/local/cuda-13.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-13.0/lib64:${LD_LIBRARY_PATH:-} export USE_CUDA=1 export USE_CUDNN=0 export USE_NCCL=0 export USE_DISTRIBUTED=0 export BUILD_TEST=0 export TORCH_CUDA_ARCH_LIST="8.0;8.6;8.9;9.0;10.0;12.0" export MAX_JOBS=8 export PYTORCH_BUILD_VERSION=2.11.0 export PYTORCH_BUILD_NUMBER=1 python3 -m pip wheel --no-build-isolation --no-deps --wheel-dir dist .

  构建产物示例如下：

  1 torch-2.11.0-cp311-cp311-linux_aarch64.whl

4. 安装本地构建产物。
  1 2 3 4 5 6 dnf install -y \ --setopt=install_weak_deps=False \ --setopt=tsflags=nodocs \ openblas numactl-libs python3 -m pip install --force-reinstall --no-deps dist/torch-2.11.0-*.whl