解決TFServing grpc模式 時報No module named 'tensorflow_serving'

調試TFServing  grpc模式 ，部署相關模型後報如下問題

File "resnet_client_grpc.py", line 27, in <module>
    from tensorflow_serving.apis import predict_pb2
ImportError: No module named 'tensorflow_serving'

解決

  pip install  tensorflow-serving-api=='1.12.0'

  一定要寫 好對應cuda 匹配的版本號，不寫默認會把tensorflow 和Tensorboard升到1.14.0

參考  https://github.com/tensorflow/serving/issues/961