Jetson Orin NX（Arm aarch64） Pytorch 安装

网上好多教程都有点差错，弄不下来，最终还是whl安装解决了。

注意一定要选择aarch64

下载网址：PyTorch for Jetson - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums

orin nx使用sdkmanager 刷机
cuda11.4 python3.8.10，我选择了PyTorch1.11.0下载

 不知道为什么，这里下载后文件名是乱码（1.12.0正常），注意要给它改名后再pip。
改名为绿色那一排字

下载whl文件，改名后，终端cd到文件的位置后：

pip install torch-1.11.0-cp38-cp38-linux_aarch64.whl

安装后，在终端

python3
import torch
torch.cuda.is_available()

返回是True

torchvison版本对应

下载对应版本，可以在这里找，同样注意选择aarch64：https://torch.kmtea.eu/whl/stable.html

这里也可以下载pytorch，我注意到有2.0版本的

torch · PyPI

有错误麻烦指正