基本流程
- docker pull tensorflow serving官方镜像
- 生成tensorflow serving可直接读取的模型格式
- 编写配置文件
- 启动tensorflow serving镜像
1、docker下载
- CPU版本
docker pull tensorflow/serving:latest
- GPU版本
docker pull tensorflow/serving:latest-gpu
2、模型格式转换
tensorflow serving支持pb模型,部署多模型所需的相关模型结构目录如下:

我用的是kashgari框架,其提供直接保存pb模型的方法kashgari.utils.convert_to_saved_model

-----model.save:保存在reberate_classification_13文件夹下的h5模型

-----convert_to_saved

本文档详细介绍了如何使用TensorFlow Serving进行多模型部署,包括从Docker镜像下载、模型格式转换、配置文件编写到服务启动的全过程。重点讲解了kashgari框架如何生成可供TensorFlow Serving读取的pb模型,并提供了CPU和GPU版本的启动命令。
2485

被折叠的 条评论
为什么被折叠?



