兄弟们,最近搞了几个大模型部署项目,从单机推理到分布式架构,一路踩坑不少。来分享点干货。
先说推理框架选型。别一上来就上vLLM、TensorRT-LLM这些,得先搞清楚模型参数量、精度要求和并发量。比如7B模型,单卡A100用vLLM做FP16推理,256并发基本能吃满;但换成70B,就得考虑TP+PP的分布式方案,还得搭配KV Cache优化。
再说数据预处理。很多人忽略这个,结果模型跑起来才发现batch size调不上去。Tokenizer的并行化、数据管道的预取机制、动态batching策略,这些都得提前规划好。特别是用PyTorch做推理时,DataLoader的num_workers设置直接影响吞吐量。
最后说监控。别等到模型卡死才去查日志。把GPU显存、利用率、推理延迟、QPS都打到Grafana面板上,设好告警阈值。我见过最坑的是模型加载阶段内存泄漏,跑一周才爆,没监控根本发现不了。
问题抛一个:你们在实际部署中,遇到过哪些奇葩的架构问题?比如分布式通信瓶颈、显存碎片化之类的,来评论区聊聊。 |