关于深度学习:Tensorflow-serving-with-docker

47次阅读

共计 347 个字符,预计需要花费 1 分钟才能阅读完成。

本次教程的目标是率领大家看下如何用 Docker 部署深度学习模型的

第一步咱们须要 pull 一个 docker image

sudo docker pull tensorflow/serving

如上图所示,执行 pull 之后,咱们看到本地曾经存在 tensorflow/serving:latest

接下来咱们 clone 一个仓库

git clone https://github.com/tensorflow/serving

上图中的 saved_model_half_plus_two_cpu 就是咱们想要部署的模型

而后咱们能够间接运行以下命令实现部署

运行后果如下图所示,咱们能够看到 Exporting HTTP/REST API at:localhost:8501,那么就代表着部署胜利了

接下来咱们能够进行预测,返回的后果也能对的上

正文完
 0