一、问题描述
ubuntu 18..04 x86主机下,将tensorflow(1.14.0) checkpoint模型文件导出为pb模型文件;将pb模型文件部署在nvidia tx2 aarch64(tensorflow==1.13.1)上进行推理,出现如标题所示报错。
二、问题原因
1、pb模型导出环境的cpu架构与部署环境cpu架构不一致。
2、网上有说因为tf版本不一致所致,尝试将tf版本设为一致后,仍然会报错
三、解决方法
在部署环境下将tf checkpoint模型文件导出为pb模型文件,再进行pb模型部署。