NVIDIA系列Jetson NX开箱跑tensorRT+yolov3

1 篇文章 0 订阅
1 篇文章 0 订阅

买来的Xavier NX是不带SD卡的,记得提前买个,别机子到了,SD卡(64G起)还没到

买来第一步,那显然是得刷系统,这我就贴个别人帖子了 刷机帖子  毕竟我也是在那下的,如果没百度云会员,那下载速度杠杠滴,可以留言,我有时间的时候把东西通过邮箱发

装机完后,先跑个自带的tensorRT+yolov3,顺便测测速

上来先跟着readme走,走到

python3 -m pip install -r requirements.txt

直接卡住,各种报错,后来找到原因后,就是protobuf和onnx这两个版本问题,他们两哥们的版本不对,后面的

python3 onnx_to_tensorrt.py

也不可能过得去,然后就围着这搞了我两天的时间,最后在论坛里面找到了解决方案,我最终的版本

protobuf的版本是3.10.1,onnx的版本是1.6.0,然后就可以跑demo了,顺便测下速

 

一张图差不多15ms就完事了,跑完demo我想让他int8量化后,保存量化的模型并再识别同张图片,看看时间差多少?然后这部又卡了,因为demo里面只给了测速,没给保存int8量化的方法,又开始一顿搜索,终于找到了如下指令

sudo /usr/src/tensorrt/bin/trtexec --onnx=yolov3.onnx --int8 --batch=1 --device=0 --saveEngine=yolov3.engine --verbose

一定记得前面加sudo,因为装系统的时候,已经把这部分demo默认丢到了root下,不加sudo,这个指令不会保存模型,我跑了n遍才恍然大悟 得加sudo,跑完int8的结果就不暴了,弄完之后发现NVIDIA系列还是牛逼,性价比几乎完美,为算法落地提供了不错的平台

 

评论 33
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值