在使用TensorFlow Serving的 gRPC 模式进行客户端与服务器通信时,有时会报 No module named ‘tensorflow_serving’ 错误,如下:
这时我们就需要自己手动安装 tensorflow-serving-api
库:
pip3 install --user tensorflow-serving-api=='1.14.0' # --user 可选,表示仅代表该用户可用
注: 建议安装 tensorflow-serving-api
时指定安装版本,安装版本最好与自己电脑中的TensorFlow及CUDA版本对应,否则会自动安装(升级)TensorFlow及Tensorboard等依赖库,影响其他程序环境。 如需查看CUDA及cuDNN版本,可查看:https://blog.csdn.net/weixin_40922744/article/details/103029070