如何在TF网站中设置Tensorflow Serving?

时间:2019-01-29 15:14:45

标签: python tensorflow

我正在尝试根据here

来学习和设置TF服务

我使用retrain.py文件训练了模型,所有必需的模型文件都在“ saved_models”文件夹中 enter image description here

现在,我要更新服务器命令

    # Start TensorFlow Serving container and open the REST API port
docker run -t --rm -p 8501:8501 \
   -v "$TESTDATA/saved_model_half_plus_two_cpu:/models/half_plus_two" \
   -e MODEL_NAME=half_plus_two \
   tensorflow/serving &

我很难理解如何将示例路径更改为当前设置。 我不明白“ / models / half_plus_two”是如何到达那里的,因为它在示例文件夹中不可用。

这就是我在这里所做的

docker run -t --rm -p 8501:8501 -v "c:/tmp/saved_models:/models/1" -e MODEL_NAME=c:\tmp\saved_models\1\saved_model.pb

但是失败

C:\tmp\serving>docker run -t --rm -p 8501:8501 -v "c:/tmp/saved_models:/models/1" -e MODEL_NAME=c:\tmp\saved_models\1\saved_model.pb
"docker run" requires at least 1 argument.
See 'docker run --help'.

Usage:  docker run [OPTIONS] IMAGE [COMMAND] [ARG...]

Run a command in a new container

我们非常感谢您的帮助。

1 个答案:

答案 0 :(得分:0)

您是否已将docker拉到tensorflow /服务映像?另外,您需要为docker运行一个“映像”(您之前拉过的映像)。

docker pull tensorflow/serving