Jetson Orin TorchVision whl 自编译指南

1
git clone --recursive --branch v0.22.0 https://github.com/pytorch/vision torchvision
2
cd torchvision
3
sudo apt-get update && sudo apt-get install -y libjpeg-dev libpng-dev libwebp-dev libavcodec-dev libavformat-dev libswscale-dev ffmpeg

1
source ../pytorch/.venv/bin/activate
2
uv pip install numpy pillow

1
export CPATH="/usr/include/aarch64-linux-gnu:/usr/local/cuda/include:$CPATH"
2
export LIBRARY_PATH="/usr/lib/aarch64-linux-gnu:/usr/local/cuda/lib64:$LIBRARY_PATH"
3
export LD_LIBRARY_PATH="/usr/lib/aarch64-linux-gnu:/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
4
export FORCE_CUDA=1
5
export TORCH_CUDA_ARCH_LIST="8.7"
6

7
python3 setup.py bdist_wheel

1
uv pip install ./dist/torchvision-0.22.0+9eb57cd-cp312-cp312-linux_aarch64.whl

1
python -c "
2
import torch
3
import torchvision
4
print(f'Torchvision Version: {torchvision.__version__}')
5
input_tensor = torch.rand(5, 4).cuda()
6
scores = torch.rand(5).  cuda()
7
try:
8
    torchvision.ops.nms(input_tensor, scores, 0.5)
9
    print('✅ CUDA Operators: SUCCESS')
10
except Exception as e:
11
    print(f'❌ CUDA Operators: FAILED, error: {e}')
12
from torchvision.io import image
13
print('✅ Basic Image IO: Functional')
14
"