Jetson Nano如何安装 onnxruntime gpu Python包
创始人
2024-02-21 16:42:33
0

Jetson Nano 可以直接 pip 安装 cpu 版本的 onnxruntime
用 pypi 的源,直接安装 onnxruntime-gpu 或者 onnxruntime_gpu 都会报找不到对应的架构 aarch64

本来以为要用官方库编译一下 https://github.com/microsoft/onnxruntime

幸运的是,NVIDIA官方帮咱编译了,参考issue:
https://github.com/microsoft/onnxruntime/issues/6124

在 https://elinux.org/Jetson_Zoo#ONNX_Runtime 中:

根据 你的JetPack 版本以及 Python 版本,选择对应的 whl 包
在这里插入图片描述
以下是一个简单的安装 demo:

# 从以上提到的链接中下载 pip whl 包
$ wget https://nvidia.box.com/shared/static/jy7nqva7l88mq9i8bw3g3sklzf4kccn2.whl -O onnxruntime_gpu-1.10.0-cp36-cp36m-linux_aarch64.whl# pip 安装 whl 包
$ pip3 install onnxruntime_gpu-1.10.0-cp36-cp36m-linux_aarch64.whl

这次偷了懒,哪天再遇到问题,一定要编译一下[捂脸笑哭]

若安装之后报了这个错:

AttributeError: module 'onnxruntime' has no attribute 'InferenceSession'

可能是因为同时安装了 gpu 版本和 cpu 版本,需要卸载掉 cpu 版本

python3 -m pip uninstall onnxruntime

同时可能需要重装 onnxruntime-gpu

上一篇:【Vue】Vuex-store

下一篇:(二) Docker安装

相关内容

热门资讯

【PdgCntEditor】解... 一、问题背景 大部分的图书对应的PDF,目录中的页码并非PDF中直接索引的页码...
修复 爱普生 EPSON L4... L4151 L4153 L4156 L4158 L4163 L4165 L4166 L4168 L4...
在Word、WPS中插入AxM... 引言 我最近需要写一些文章,在排版时发现AxMath插入的公式竟然会导致行间距异常&#...
监控摄像头接入GB28181平... 流程简介将监控摄像头的视频在网站和APP中直播,要解决的几个问题是:1&...
protocol buffer... 目录 目录 什么是protocol buffer 1.protobuf 1.1安装  1.2使用...
Windows10添加群晖磁盘... 在使用群晖NAS时,我们需要通过本地映射的方式把NAS映射成本地的一块磁盘使用。 通过...
牛客计算器的改良(Python... 文章目录1.题目描述2.输入描述:3.输出描述:4.示例15.分析6.代码7.结语 链接࿱...
【前端】‘??‘与‘||‘有什... 0 问题 经常写const data = res.data.a ?? ''或者const d...
正大杯|市调大赛|2023备赛... 关键信息 同时随着精细化养宠趋势的深入,宠物消费类目日渐丰富。 本报告通过 Niuco...
文本生成视频Make-A-Vi... Meta公司(原Facebook)在今年9月29日首次推出一款人工智能系...