如何在kubeflow中指定张量流的版本?

时间:2018-07-30 07:57:55

标签: tensorflow-serving kubeflow

我正在尝试使用tensorflow服务来服务模型。当我尝试使用kso​​nnet应用服务组件时,我看到在kubernetes(gke)上创建的工作负载正在使用tensorflow 1.7(gcr.io/kubeflow-images-public/tensorflow-serving-1.7:v20180604-0da89b8a)。

有没有一种方法可以指定在Kubeflow中提供哪个版本的tensorflow?

2 个答案:

答案 0 :(得分:2)

TensorFlow的版本由您使用的TensorFlow Serving Docker映像确定。如果您将Kubeflow ksonnet原型用于TFServing,则可以使用参数modelServerImage来设置图像。

答案 1 :(得分:2)

此外,在创建服务原型后,您可以运行ks show <env_name> -c <prototype_name>来查看将要部署的实际YAML文件。这可能使您对实际部署的内容更具可读性,然后可以更深入地了解jsonnet,以了解如何使用您所设置的参数设置特定变量(在本例中为引用tf服务版本的图像)在params.jsonnet中指定。