No matching distribution found for onnxruntime-gpu

我尝试在Xavier NX上训练模型, 利用Pytorch训练yolo v5s, 训练完毕之后, 将模型转化成TensorRT的推理引擎的时候, 需要安装onnxruntime-gpu

在Jetpack 5.0 DP上安装onnxruntime-gpu时, 出现了错误:

pip3 install onnxruntime-gpu

ERROR: Could not find a version that satisfies the requirement onnxruntime-gpu (from versions: none)
ERROR: No matching distribution found for onnxruntime-gpu

其实就是目前公开源中的版本不支持Jetson平台, 可以直接访问

Jetson Zoo - eLinux.org

下载对应的版本

 然后重新使用pip3来安装就好:

pip3 install onnxruntime_gpu-1.11.0-cp38-cp38-linux_aarch64.whl

Processing ./onnxruntime_gpu-1.11.0-cp38-cp38-linux_aarch64.whl
Collecting flatbuffers
  Downloading flatbuffers-2.0-py2.py3-none-any.whl (26 kB)
Requirement already satisfied: numpy>=1.22.3 in /home/hekun/.local/lib/python3.8/site-packages (from onnxruntime-gpu==1.11.0) (1.22.4)
Requirement already satisfied: protobuf in /home/hekun/.local/lib/python3.8/site-packages (from onnxruntime-gpu==1.11.0) (4.21.0)
Installing collected packages: flatbuffers, onnxruntime-gpu
Successfully installed flatbuffers-2.0 onnxruntime-gpu-1.11.0