Google Cloud ML Engine:使用Tensorflow 1.10?

时间:2018-11-06 01:31:33

标签: tensorflow

我已经使用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创建模型版本?

谢谢!

1 个答案:

答案 0 :(得分:0)

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

  

注意:AI平台使用的默认AI平台运行时版本   培训和       预测API是1.0版。当您提交培训工作要求并部署模型时,       您必须指定所需的运行时版本。否则,AI平台将使用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