把SSD-tensorflow移植到Jetson-TX2经验总结

来源:互联网 发布:工业大数据 质量管理 编辑:程序博客网 时间:2024/05/29 19:00

最近由于项目的需要,需要把SSD-tensorflow 的代码移植到Jetson-TX2上来看看效果,结果还是不尽人意,我们测试的结果是每秒只能2帧左右,确实很慢,不过精度还是可以的。我这里把移植的过程分享出来。

首先需要装一些东西,命令如下:

由于可能pip库里面没有很多包,一般我建议采取apt-get来安装

sudo apt-get install python-pipsudo pip install matplotlibsudo apt-get install python-scipysudo pip install graphviz jupytersudo pip3 install --upgrade pipsudo pip install --upgrade pip enum34

接着是安装cuda和cudnn,如果一开始刷机的时候已经装好cuda了,那就只需要装cudnn了,cudnn的安装跟桌面PC端的一样,安装过程请参考我的博客:

cudnn安装

下面是一些小技巧,用得上可以用一下:

判断cuda的存在nvcc -Vubuntu Linux系统简单易用的上传下载命令rz和szsudo apt-get install lrzsz

然后需要装opencv和tensorflow,详细教程请参考我的博客:

opencv安装教程
Tensorflow安装教程


然后把你自己的代码远程传送到tx2上,就可以像在PC端上一样运行了。



原创粉丝点击