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