一、PyTorch安装
1.1 下载Jetson版本的PyTorch
Jetson平台的PyTorch安装包不是通过PyTorch官网下载的，而是由NVIDIA的PyTorch for Jetson网站提供的，Jetson的L4T版本与PyTorch版本有对应要求，根据Jetson平台的系统版本不同，提供不同的安装包。
根据下图可知JetPack5.1.2(L4T R35.4)版本对应的PyTorch版本为v2.1.0。
1.2 安装PyTorch依赖
sudo apt-get install libopenblas-dev
1.3 安装PyTorch
pip命令直接安装（注意不要用sudo命令，安装到user下就行，ubuntu20.04已经移除了python2，只有python3）。
pip install torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
1.4 测试PyTorch
安装完成后，查看是否支持CUDA及CUDA的版本。
python
import torch
print(torch.cuda.is_available())
print(torch.version.cuda)
print(torch.version.cuda)实测中发现可能需要等一会才能执行，可能需要系统配置需要时间，可以尝试重启下。
二、安装Torchvision
2.1 下载Torchvision
Torchvision版本与PyTorch版本也是存在对应关系，可在其官网查询，如下图所示，由于之前安装的PyTorch版本为2.1，因此这里Torchvision的版本必须为0.16。
通过github下载torchvision，这里我选择的是v0.16.2版本。
git clone --branch v0.16.2 https://github.com/pytorch/vision torchvision
2.2 安装依赖
下面这些不确定是否有用，但实际安装过程中还是安装了。
sudo apt-get -y update;
sudo apt-get -y install autoconf bc build-essential g++-8 gcc-8 clang-8 lld-8 gettext-base gfortran-8 iputils-ping libbz2-dev libc++-dev libcgal-dev libffi-dev libfreetype6-dev libhdf5-dev libjpeg-dev liblzma-dev libncurses5-dev libncursesw5-dev libpng-dev libreadline-dev libssl-dev libsqlite3-dev libxml2-dev libxslt-dev locales moreutils openssl python-openssl rsync scons python3-pip libopenblas-dev
sudo apt-get install libopenblas-base libopenmpi-dev libomp-dev
下面这些依赖是必须安装的
sudo apt-get update
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
2.3 编译安装
开始编译torchvision，注意不要用sudo，安装到user用户下就行。
cd torchvision
export BUILD_VERSION=0.16.2
python3 setup.py install --user
安装完成
2.4 测试torchvision
注意不要在下载的torchvision路径下import torchvision，否则会报warning。
python
import torch
import torchvision
print(torchvision.__version__)
参考：https://zhuanlan.zhihu.com/p/632052753
https://blog.csdn.net/qq_39422817/article/details/134611876
https://blog.csdn.net/qq_45679491/article/details/130384809
https://blog.csdn.net/WOSHIRENXIN/article/details/127415609
https://blog.csdn.net/WOSHIRENXIN/article/details/127415609
