前言
项目需要在tx2上用tensorflow加载已训练好的checkpoint文件,但是由于之前tx2上装的tensorflow1.3,而checkpoint文件是服务器上tf1.8生成的,加载时出现了生成器和解析器不匹配的情况,所以需要在tx2上重新安装tf1.8。
而安装tf1.8需要对应版本的cuda9.0,所以我们还需要重新刷机jetpack3.2,刷机教程之前的博客有:
https://blog.csdn.net/u014796085/article/details/84239971
tensorflow1.8安装
下面主要讲tf1.8的安装过程:
我用的是官方已经编译好的适配于tx2的tf1.8(官方github:https://github.com/NVIDIA-AI-IOT/tf_trt_models),所以应该是只需要安装一些必须的python库,再直接安装tf1.8编译文件中的install脚本就可以了。至于java和bazel这些是否需要装,我是先装了的,不装的话,能否成功安装tf1.8,不太清楚,大家可以试下然后评论告诉我。
如果需要自己在tx2上编译tf1.8,java和bazel应该是必须要装的。主要可以参考以下网址:
https://www.colabug.com/4607320.html
安装python库时,有些需要跳过pip,直接apt安装:
https://blog.csdn.net/jonado13/article/details/83933453
opencv3.4安装
最后,jetpack3.2自带的opencv是2.x版本的,安装opencv3.4可以参考:
https://jkjung-avt.github.io/opencv3-on-tx2/