Tensorflow 服务:get_model_status 缺少 'utils'

Tensorflow Serving: get_model_status is missing the 'utils'

我正在尝试使用 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 服务,我会得到目录,所以我想可以修复它来制作一个轮子,但我还是很好奇。

提前致谢。

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

安装 .whl 文件

这个问题已经 fixed,应该会在 API 的下一个版本中解决。