Skip to content

Commit

Permalink
use tf serving image (kubeflow#1370)
Browse files Browse the repository at this point in the history
* use tf serving image

* fix command

* use 1.8

* use 1.9

* use 1.8
  • Loading branch information
lluunn authored and k8s-ci-robot committed Aug 17, 2018
1 parent de14bbc commit 398e38d
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions kubeflow/tf-serving/tf-serving.libsonnet
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,8 @@
// in which case the image used will still depend on whether GPUs are used or not.
// Users can also override modelServerImage in which case the user supplied value will always be used
// regardless of numGpus.
defaultCpuImage: "gcr.io/kubeflow-images-public/tensorflow-serving-1.7:v20180604-0da89b8a",
defaultGpuImage: "gcr.io/kubeflow-images-public/tensorflow-serving-1.6gpu:v20180604-0da89b8a",
defaultCpuImage: "tensorflow/serving:1.8.0",
defaultGpuImage: "tensorflow/serving:1.10.0-gpu",
modelServerImage: if $.params.numGpus == 0 then
$.params.defaultCpuImage
else
Expand Down Expand Up @@ -114,8 +114,10 @@
name: $.params.name,
image: $.params.modelServerImage,
imagePullPolicy: "IfNotPresent",
args: [
command: [
"/usr/bin/tensorflow_model_server",
],
args: [
"--port=9000",
"--model_name=" + $.params.modelName,
"--model_base_path=" + $.params.modelPath,
Expand Down

0 comments on commit 398e38d

Please sign in to comment.