本站含联盟链接。作为 eBay 合作伙伴,你通过本站链接购买时,我们可能获得佣金。作为 eBay 合作伙伴,我们可能获得佣金。了解详情

收录显卡57价格追踪49 天近 24h 价格数据43,303 条24h 降价最多Tesla V100 16GB-12.3%24h 涨价最多A100 80GB PCIe+9.4%更新于
¥ CNY

Mistral Nemo 如何本地部署?

Mistral NeMo 是 Mistral AI 和 NVIDIA 在 2024 年 7 月合作发布的 12B 模型,官方定位是 Mistral 7B 的直接替代。短对话很轻,4-bit 在 12GB 显卡上就够;但它的上下文很吃显存,每多 32K 约多 5 GB,开满 128K 要约 30 GB。

Mistral Nemo 模型信息

  • 4-bit 最低显存10 GB · 8K
  • 参数量12.3B
  • 上下文上限128K
架构
稠密
许可证
Apache-2.0
发布日期
2024年7月17日
厂商
Mistral AI
Ollama 标签
mistral-nemo:12b
小版本
2407
层数
40
隐藏维度
5,120
KV 头
8
头维度
128

Mistral Nemo 本地部署需要多少显存?

量化权重体积GB8K 上下文GB32K 上下文GB128K 上下文(最大)GB
1-bit3.1估5.59.525.6
2-bit5.4GGUF7.811.827.9
3-bit7.1GGUF9.513.529.6
4-bit7.5GGUF9.913.930
5-bit8.7GGUF11.115.131.2
8-bit13GGUF15.419.435.5
BF16/FP1624.5GGUF26.930.947

详细计算规则可在数据方法页查看。

单卡能跑 Mistral Nemo 吗?

单卡没问题,12GB 的卡跑 4-bit 加 8K 上下文(约 9.9 GB)就够。它吃显存的地方在上下文,每多开 32K 约多占 5 GB,所以 16GB 的卡最好留在 4-bit,换成 8-bit 上下文就很紧了。兼顾质量和长度的话,24GB 的 RTX 3090 用 8-bit 能开到约 69K,速度也流畅;要开满 128K,至少得上 32GB 的 RTX 5090。好在它的新分词器写中文更省,官方说比 Mistral 之前的模型少用约 30% 的 token,同样的上下文能装下更多中文。

闲鱼
设备价格速度 t/s最大上下文
Tesla V100 16GB16 GB¥1,2505 条挂单75.651K
Tesla V100 32GB32 GB¥4,2004 条挂单75.6128K
GeForce RTX 5060 Ti 16GB16 GB¥5,17514 条挂单39.851K
Radeon RX 7900 XTX24 GB¥5,69412 条挂单80.1102K
GeForce RTX 309024 GB¥8,50013 条挂单78.3102K
Ryzen AI Max+ 395 128GB整机 · 128 GB¥19,00016 条挂单23.3128K
GeForce RTX 409024 GB¥25,0006 条挂单83.6102K
RTX A600048 GB¥32,0007 条挂单65.6128K
DGX Spark 128GB整机 · 128 GB¥43,00041 条挂单24.8128K
MacBook Pro M5 Max 128GB整机 · 128 GB¥45,69410 条挂单53.4128K
Mac Studio M5 Ultra 96GB整机 · 96 GB¥46,00015 条挂单99.4128K
GeForce RTX 509032 GB¥51,99914 条挂单136.3128K
RTX PRO 600096 GB¥125,0006 条挂单136.3128K

只列常见的显卡和整机,50B 以下和以上的模型各一组,按价格从低到高。

本地部署 Mistral Nemo 要注意什么?

  1. 温度要调低

    官方特别提醒,NeMo 比之前的 Mistral 模型需要更低的温度,建议 temperature 0.3。很多聊天前端默认 0.7 以上,记得手动改。

  2. 量化选哪一档

    官方说训练时就按量化做过适配,用 FP8 推理不掉精度。GGUF 的 4-bit 约 7.5 GB、5-bit 约 8.7 GB、8-bit 约 13.0 GB:12GB 卡用 4-bit 或 5-bit,16GB 以上再上 8-bit。

  3. 中文更省 token

    NeMo 换了新的 Tekken 分词器,官方称中文和代码的压缩效率比之前高约 30%,也就是同样 128K 的上下文能装进更多中文。

  4. 用什么框架

    llama.cpp、vLLM 直接支持,Ollama 标签是 mistral-nemo:12b。它用的是标准结构,原来跑 Mistral 7B 的工具链基本不用改。

启动命令

Ollama

  • Windows
  • macOS
  • Linux
终端
ollama run mistral-nemo:12b

llama.cpp

  • NVIDIA
  • AMD
  • Mac
  • CPU
终端
llama-server -hf bartowski/Mistral-Nemo-Instruct-2407-GGUF:Q4_K_M \
  -c 8192

vLLM

  • Linux
  • NVIDIA
  • AMD
终端
vllm serve mistralai/Mistral-Nemo-Instruct-2407

MLX

  • Apple 芯片

Mac 上先 pip install mlx-lm,这条命令会从 Hugging Face 下载 MLX 权重,并启动兼容 OpenAI 接口的服务。

终端
mlx_lm.server --model mlx-community/Mistral-Nemo-Instruct-2407-4bit

量化文件名以模型库当周抓取为准,口径见数据方法。 数据方法

Mistral 家族的其他模型

模型总参数激活参数4-bit 模型大小
Mistral 7B v0.3稠密7.3B7.3B4.4 GB
Ministral 3 8B稠密8.4B8.4B5.2 GB
Ministral 3 14B稠密13.5B13.5B8.2 GB
Mistral Small 3.2 24B稠密24B24B14.3 GB

关于 Mistral