讲师中心 微信公众号
AI工具推荐 视频效率加速

如何用Python TensorFlow将模型部署为API_通过TF Serving容器化解决

酷明姑娘_4444

酷明姑娘_4444

发布时间:2026-04-30 09:25:19

|

1014人浏览过

|

来源于php中文网

原创

TensorFlow Serving生产部署应采用gRPC服务+FastAPI协议转换方案,SavedModel须用tf.keras.models.save_model显式指定tf格式并确保版本目录结构正确,启动需同时配置--rest_api_port和--enable_batching,gRPC客户端需版本严格匹配且stub全局复用。

如何用python tensorflow将模型部署为api_通过tf serving容器化解决

TensorFlow Serving 是目前最稳妥的生产级部署方案,但直接用它暴露 HTTP API 会踩坑——它原生只提供 gRPC 接口,HTTP 需额外加一层代理或改用 tensorflow-serving-api 的 Python 封装,而后者在高并发下不稳定。真正可靠的做法是保留 TF Serving 的 gRPC 服务,再用轻量 Web 层(如 Flask/FastAPI)做协议转换。

导出 SavedModel 格式必须用 tf.keras.models.save_model,别用 model.save 默认路径

TF Serving 只认标准 SavedModel 目录结构(含 saved_model.pb 和 variables/ 子目录)。model.save('my_model') 在 TF 2.x 默认行为可能写成 HDF5(.h5),或路径里混入时间戳导致版本管理混乱。

  • 显式指定格式:
    tf.keras.models.save_model(model, 'models/my_model/1', save_format='tf')
    注意末尾的 1 是版本号目录,TF Serving 按数字升序加载最新版
  • 确保目录权限为 755,且 variables/ 下文件可读——容器内常因挂载方式丢失执行权限,启动时报 Failed to get matching files
  • 验证导出是否有效:saved_model_cli show --dir models/my_model/1 --all,确认 signature_def 里有 serving_default 且 input tensor 名与你后续请求一致

启动 TF Serving 容器时,--rest_api_port 不等于开启完整 HTTP 服务

TF Serving 从 2.10 开始内置实验性 REST API,但仅支持基础 predict 请求,不支持 model status、version switching 等管理接口,且默认关闭。关键参数不是 --rest_api_port 单独生效。

testing-python
testing-python

使用pytest编写和评估有效的Python测试。适用于编写测试、审查测试代码、调试测试失败或提高测试覆盖率。

下载
  • 必须同时加 --enable_batching=true(否则 batch size=1 时延迟飙升)和 --rest_api_port=8501
  • 容器启动命令示例:
    docker run -p 8501:8501 -p 8500:8500 --mount type=bind,source=/path/to/models,target=/models/my_model -e MODEL_NAME=my_model -t tensorflow/serving --rest_api_port=8501 --enable_batching=true
  • 常见错误:用 curl http://localhost:8501/v1/models/my_model 返回 404——检查 MODEL_NAME 是否与目录名完全一致(区分大小写),且模型版本目录名是纯数字

用 FastAPI 封装 gRPC 调用比直连 REST API 更可控

TF Serving 的 REST 接口对输入格式极其敏感:instances 字段必须是 list of dict,哪怕单样本也要包一层;图像 base64 编码后需额外处理;batch 维度缺失时自动广播易引发隐式 bug。gRPC + Python stub 则类型明确、性能稳定。

立即学习“Python免费学习笔记(深入)”;

  • 安装客户端:pip install tensorflow-serving-api(版本必须与服务端 TF Serving 严格一致,如服务端是 2.15.0,客户端不能是 2.16.0)
  • 关键代码片段:
    from tensorflow_serving.apis import predict_pb2, prediction_service_pb2_grpc
    import grpc
    stub = prediction_service_pb2_grpc.PredictionServiceStub(grpc.insecure_channel('localhost:8500'))
    request = predict_pb2.PredictRequest()
    request.model_spec.name = 'my_model'
    request.model_spec.signature_name = 'serving_default'
    request.inputs['input_1'].CopyFrom(tf.make_ndarray(tf.constant([[1.0, 2.0]])))  # 注意 input 名要和 saved_model_cli 输出一致
  • FastAPI 路由中不要在每次请求里新建 stub——gRPC channel 是线程安全的,应全局复用;否则高并发下 fd 耗尽,报错 OS Error: Too many open files

真正难的是模型热更新和流量灰度——TF Serving 支持多版本共存,但切换路由依赖外部负载均衡或自定义 ModelServer 配置;而 Python 层做 A/B 测试时,tensor 输入预处理逻辑若放在 FastAPI 里,就和模型耦合了,下次换框架就得重写。这些细节比“怎么跑起来”更消耗上线时间。

热门AI工具

更多
UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1671

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

4204

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1669

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

24437

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2987

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3007

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1163

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

596

2023.08.10

FrankenPHP集成Laravel详细教程
FrankenPHP集成Laravel详细教程

本专题提供FrankenPHP集成Laravel的详细配置指南,全面解析运行原理、开发环境搭建、Caddyfile配置、Octane工作模式、数据库连接、队列任务、定时任务和生产环境优化,解决部署过程中常见的报错与兼容性问题。

0

2026.10.08

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn