1.查看jetpack版本：sudo jtop

自帶的就有cuda11.4和cudnn8.X以及python3.8，我的cudnn就沒有是后期自己安裝的

2.安裝torch

PyTorch for Jetson - Announcements - NVIDIA Developer Forums

選擇對應的cuda版本和torch版本，我下載的是：torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64

直接用pip install torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64 安裝即可

3.如果系統自帶的沒有cudnn則需要手動安裝：

cuDNN Archive | NVIDIA Developer

選擇對應的版本，我選的是 libcudnn8_8.2.2.26-1+cuda11.4_arm64.deb下載

安裝：

sudo dpkg -i  libcudnn8_8.2.2.2