返回顶部
7*24新情报

Llama 3.1 405B开源实测:推理快但吃显存,部署避坑指南

[复制链接]
jack143 显示全部楼层 发表于 前天 15:20 |阅读模式 打印 上一主题 下一主题
兄弟们,Meta前几天甩出Llama 3.1 405B开源模型,号称性能对标GPT-4。我连夜在A100上跑了下,说点干货。

实测下来,405B的推理速度确实比上一代快30%左右,长上下文(128K)处理也稳,不会像某些模型到一半就崩。但别高兴太早——它显存需求炸裂,单卡A100 80G只能跑4-bit量化版,全精度部署至少8卡起步。想玩的话,推荐用vLLM或TGI,搭配FlashAttention-2,延迟能降到3秒内。

另外,RAG场景下,它跟LangChain集成不错,但注意别用默认chunk size,调成512 token效果更好。有个坑:中文分词偶尔抽风,建议先用Jieba预处理。

最后,别信官方吹的“零样本完胜”,我测了数学推理(GSM8K),它跟GPT-4还有5%差距,但开源能这样已经算卷王了。想省钱的,直接去HuggingFace下4-bit GGUF,单卡RTX 4090也能跑。

老规矩,部署遇到OOM不要慌,先调low_cpu_mem_usage=True。评论区见。
回复

使用道具 举报

default_avator1
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

Archiver·手机版·闲社网·闲社论坛·羊毛社区· 多链控股集团有限公司 · 苏ICP备2025199260号-1

Powered by Discuz! X5.0   © 2024-2025 闲社网·线报更新论坛·羊毛分享社区·http://xianshe.com

p2p_official_large
快速回复 返回顶部 返回列表