常规的pip install 安装torch onnx在 Xavier上面不太行
因此参考在Jetson平台上安装pytorch
Installing PyTorch for Jetson Platform - NVIDIA Docs
Torch安装解决方案:
因为我的Jet pack是4.6.1 所以参考下面链接查找对应的Jet pack 对应torch weel版本文件PyTorch for Jetson - Jetson Nano - NVIDIA Developer Forums
下载torch 1.10.0
下载完成后 输入以下命令安装 torch 1.10.0(Jet pack 1.10.0, pyhton3.6)
pip3 install torch-1.10.0a0+git36449ea-cp36-cp36m-linux_aarch64.whl
ONNX 安装
参考下面链接