Jetson Orin PyTorch whl 自编译指南

1
git clone --recursive --branch v2.7.0 http://github.com/pytorch/pytorch
2
cd pytorch
3
sudo apt update && sudo apt-get install python3-pip cmake libopenblas-dev libopenmpi-dev -y

1
uv venv .venv --python 3.12
2
uv pip install -r requirements.txt
3
uv pip install scikit-build
4
source .venv/bin/activate

1
export PYTORCH_BUILD_VERSION=2.7.0
2
export PYTORCH_BUILD_NUMBER=1
3
export TORCH_CUDA_ARCH_LIST="8.7"
4

5
export USE_NCCL=0             # 关闭 Jetson 用不上的模块
6
export USE_DISTRIBUTED=0
7
export USE_QNNPACK=0
8
export USE_PYTORCH_QNNPACK=0
9

10
export CUDA_HOME=/usr/local/cuda
11
export PATH=$CUDA_HOME/bin:$PATH
12
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
13
export USE_PRIORITIZED_TEXT_FOR_LD=1
14

15
export MAX_JOBS=$(nproc)
16
python3 setup.py bdist_wheel

1
uv pip install ./dist/torch-2.7.0-cp312-cp312-linux_aarch64.whl

1
python -c"import torch
2
print(torch.cuda.is_available())"