官方的: https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html#overview__section_orin
一步一步来
sudo apt-get -y update;
sudo apt-get -y install python3-pip libopenblas-dev;
python3 -m pip install --upgrade pip;
python3 -m pip install numpy
python3 -m pip install --no-cache https://developer.download.nvidia.cn/compute/redist/jp/v511/pytorch/torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl
git clone --branch v0.15.1 https://github.com/pytorch/vision torchvision
git clone --branch v0.15.1 https://gitclone.com//github.com/pytorch/vision torchvision
cd torchvision
python3 setup.py install --user
import torch
import torchvision
torch.__version__
torchvision.__version__
torch.cuda.is_available()