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
