贝壳:基于 GPU 虚拟化扩展机器学习基础设施

背景

贝壳找房是中国领先的线上线下房产交易和服务平台。为支持公司快速增长的 AI 计划,集中式基础设施团队运营着跨所有业务部门共享的机器学习平台。

该团队为模型开发、训练和大规模推理提供端到端的计算服务,支持内部研究工作负载和面向生产的 AI 服务。随着模型采用率和请求量在整个组织中的增长,GPU 效率和工作负载隔离成为平台的关键要求。

挑战

随着贝壳机器学习计划的扩展,基础设施团队在 GPU 资源管理方面面临重大挑战:

最初,由于多云环境的复杂性和多样化的工作负载需求,整体 GPU 利用率仅为 13%,这促使基础设施团队寻求提高集群资源利用率的解决方案。

解决方案

贝壳基础设施团队以 CNCF 项目 HAMiKubernetes 为基础,设计并实施了 AIStudio,这是一个作为组织机器学习基础设施基础的智能计算平台。

Kubernetes 因其卓越的稳定性和强大的集群调度与管理能力而被选中,显著降低了大规模集群的运营复杂性和维护开销。此外,Kubernetes 与 CNCF 开源生态系统的集成使得能够无缝采用针对不同用例的各种开源解决方案,例如 HAMi。

HAMi 被选中是因为它是 AI Studio 需求最合适的 GPU 多路复用和异构计算解决方案。

团队实施了双集群方法,根据资源需求分离工作负载。

这种架构分离确保训练作业获得专用、可预测的资源,而推理服务通过内存共享实现高密度,消除了不同工作负载类型之间的资源争用,并最大化了整体基础设施效率。

成效

通过利用包括 HAMi 和 Kubernetes 在内的开源技术,基础设施团队开发的 AI Studio 取得了以下成果:

  • 大规模稳定运行

    • 每天平稳处理数千万个业务请求
    • 关键工作负载的高可用性和可靠性
    • 高负载下的一致性能
  • 多云环境中的成本效益资源管理

    • GPU 利用率近 3 倍提升(13% → 37%)
    • 为不同工作负载类型设计高效的内存分配策略
  • 关键业务工作负载的生产级可靠性

    • 强大的调度和管理能力
    • 降低运营复杂性和维护开销

HAMi 作为基础组件的成功集成展示了开源技术如何使组织实现卓越的基础设施效率。

Kubernetes 作为底层平台基础,通过其强大的调度和管理能力,实现了每天数千万业务请求和数万个 Pod 的稳定运行。通过利用 HAMi 的 GPU 多路复用和异构调度优化功能,集群的 GPU 利用率提高了近 3 倍。

未来计划

贝壳基础设施团队继续在 HAMi 和 Kubernetes 之上创新和扩展其平台,包括:

  • 采用异构设备:计划纳入华为昇腾和其他非 NVIDIA 加速器
  • 云扩展:与阿里云集成,补充现有的火山引擎和腾讯云部署
  • 混合工作负载的高级调度策略:网络拓扑感知、卡类型规范和基于 UUID 的分配

HAMi 是 CNCF 托管的开源项目与社区,核心方向是 GPU 虚拟化与异构算力调度,面向 AI 场景提升算力利用效率;其目标是实现 灵活、按需、弹性、可靠 的 GPU 虚拟化,并持续扩展对主流 AI 芯片生态的支持。

「密瓜智能(Dynamia) 」专注 GPU 虚拟化与异构算力调度,发起并主导 CNCF 开源项目 HAMi;同时基于 HAMi 提供商业发行版、企业产品与服务,帮助用户在真实业务中规模化使用相关能力

官网:dynamia.ai

邮箱:info@dynamia.ai

本文作者「Dynamia密瓜智能」,首发于:https://dynamia.ai/zh/blog/ke-holdings-ml-infrastructure-scaling

相关推荐
PILIPALAPENG21 小时前
gh:终端里的GitHub总控台,AI时代的开发者神器
前端·人工智能·后端
Csvn21 小时前
AI 辅助知识管理与学习优化
人工智能·ai编程
ji1985944321 小时前
局部线性嵌入(LLE)算法 MATLAB 实现
算法·机器学习·matlab
0X7821 小时前
Windows 上 Codex Desktop 的 Chrome 和 Computer Use 插件不可用:一次完整排查与修复
人工智能·chatgpt·ai编程
zhangfeng113321 小时前
,在slurm中也能安装ubundu了,Singularity(现叫 Apptainer)不需要root权限的容器方案,对比docker
运维·人工智能·机器学习·docker·容器
jeffer_liu21 小时前
Spring AI 生产级实战:模型选择
java·人工智能·spring boot·后端·spring·语言模型·ai编程
fan654041421 小时前
AI搜索优化中的长尾词精准布局策略:以宠物眼科关键词为例
人工智能
Deepoch21 小时前
Deepoc VLA开发板:无人机群体协同与无网络自主作业核心
网络·人工智能·算法·无人机·deepoc·具身模型开发板
美狐美颜SDK开放平台21 小时前
直播软件开发+AI美颜SDK:下一代直播平台搭建方案
大数据·人工智能·实时音视频·美颜sdk·第三方美颜sdk
王莎莎-MinerU21 小时前
Agent 时代的科学数据 API:用 Sciverse 构建可追溯的科研检索与 RAG 工作流
大数据·人工智能·gpt·aigc·个人开发