NVIDIA系列Jetson NX开箱跑tensorRT+yolov3

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

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

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

上来先跟着readme走,走到

python3 -m pip install -r requirements.txt

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

python3 onnx_to_tensorrt.py

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

NVIDIA系列Jetson NX开箱跑tensorRT+yolov3_第1张图片

NVIDIA系列Jetson NX开箱跑tensorRT+yolov3_第2张图片

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

 

NVIDIA系列Jetson NX开箱跑tensorRT+yolov3_第3张图片一张图差不多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系列还是牛逼,性价比几乎完美,为算法落地提供了不错的平台

 

你可能感兴趣的:(jetson,NX,demo,开箱)