源码编译构建

获取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