Tensorflow服务:get_model_status缺少“ utils”

时间:2018-07-30 13:06:31

标签: python status tensorflow-serving

我正在尝试使用Tensorflow服务的一些(大概)标准功能,但是我遇到了一个奇怪的问题。我正在尝试对https://www.tensorflow.org/serving/setup之后的投放模型进行一些简单的运行状况检查。

为此,我首先使用Ubuntu 18.04创建了一个简单的Docker映像(也针对16.04进行了尝试),然后安装了python3-dev(在撰写本文时为python3.6),然后下载并安装了如上所述的包括gRPC在内的必要软件包。 。由于我对从源代码构建它不感兴趣,因此我跳过了Bazel。 正在做:

pip3 install tensorflow-serving-api

可以毫无问题地安装tensorflow-serving-api(在撰写本文时为1.9.0),但是当我尝试执行以下操作时:

from tensorflow_serving.apis import get_model_status_pb2

要利用模型状态,会出现以下错误:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_serving/apis/get_model_status_pb2.py", line 17, in <module>
    from tensorflow_serving.util import status_pb2 as tensorflow__serving_dot_util_dot_status__pb2
ImportError: No module named 'tensorflow_serving.util'

/usr/local/lib/python3.6/dist-packages/tensorflow_serving没有util目录。仅使用pip安装tensorflow_serving并知道此修复程序时,还有其他人会遇到此问题吗?如果我确实从源代码构建了tensorflow服务,我会得到目录,因此我想有一个解决办法,但是我还是很好奇。

谢谢。

2 个答案:

答案 0 :(得分:0)

要解决此问题,请从给定链接下载滚轮https://pypi.org/project/tensorflow-serving-api/#files。您可以使用7-zip提取器提取它,并将其用作导入的源文件夹,或者使用.whl安装pip文件。

答案 1 :(得分:0)

此问题已fixed,应在下一版API中解决。