一、PyTorch
1、torch下载
torch版本选择由JetPack版本决定,JetPack 是 NVIDIA 为 Jetson 系列嵌入式计算平台提供的软件开发套件(SDK),输入以下两个命令均可以查看JetPack版本,我的是5.1.2
dpkg -l | grep nvidia-jetpack
sudo apt show nvidia-jetpack
官网不同JetPack版本对应的torch下载链接 Index of /compute/redist/jp, 我的JetPack版本是5.1.2,所以点击v512即可下载对应pytorch版本torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
2、torch安装
nvidia官网在Jetson 平台安装PyTorch教程: