部署到GCP Cloudml的错误模型

时间:2018-04-08 07:02:34

标签: tensorflow google-cloud-platform

我正在尝试将使用Tensorflow 1.7培训的模型部署到Google Cloud Platform上。我收到以下错误:

创建版本失败。检测到错误的错误模型:"无法加载模型:加载可服务:{名称:默认版本:1}失败:未找到:操作类型未注册' SparseFillEmptyRows' \ n \ n(错误代码:0)"

我知道Cloudml运行时预测仅支持Tensorflow 1.6,所以我尝试指定:

REQUIRED_PACKAGES = [
  'tensorflow==1.6',
]
在setup.py中

但我仍然收到相同的消息

感谢任何帮助

2 个答案:

答案 0 :(得分:0)

您需要使用TensorFlow 1.6重建模型。您无法将使用TensorFlow 1.7创建的模型部署到ML引擎。

此外,您可以将引擎运行时的版本设置为here列出的版本之一。如果您使用的是gcloud ml-engine jobs submit training,则可以使用--runtime-version标记设置版本。文档为here

答案 1 :(得分:0)

使用1.6重建并使用--runtime-version = 1.6进行部署。