首页 > 解决方案 > 谷歌云机器学习引擎:使用 TensorFlow 1.10?

问题描述

我已经使用 Tensorflow 1.10 训练了一个 Keras 模型。但是,当我在 GUI 中创建模型版本时,我最多只能在 ML Engine 上看到 Tensorflow 1.9。如果我使用 1.9,我可以得到预测,但由于版本不匹配,这让我感到担忧。

话虽如此,这里的文档似乎表明支持 1.10:https ://cloud.google.com/ml-engine/docs/tensorflow/runtime-version-list 。

如何使用 Tensorflow 1.10 创建模型版本?

谢谢!

标签: tensorflow

解决方案


您提供的链接中的文档似乎更新了以下内容:

注意:AI Platform 训练和预测 API 使用的默认 AI Platform 运行时版本是 1.0 版。当您提交训练作业请求和部署模型时,您必须指定所需的运行时版本。否则,AI Platform 使用 1.0 版。了解如何设置运行时版本。

例子:

gcloud ai-platform jobs submit training my_job \
    --module-name trainer.task \
    --job-dir gs://my/training/job/directory \
    --package-path /path/to/my/project/trainer \
    --python-version 3.5
    --region us-central1 \
    --runtime-version 1.13

推荐阅读