本地部署的硬件选型:CPU/GPU/内存/存储要求、常见配置推荐、异构计算考量
好,咱们进入第二章。说实话,硬件选型这块,我见过太多人一开始就在“买什么卡”上纠结半天。其实你想想看,选硬件就像给模型找匹马——你得先知道这匹马要拉多重的货、跑多远的路。
我个人习惯,在决定硬件之前先回答三个问题:
- 模型多大? 几B参数?还是几十B?
- 要跑多快? 是实时推理,还是离线批量处理?
- 跑多久? 训练用几天?推理一直在跑?
想清楚这些,选型才有方向。
核心部件选型:CPU、GPU、NPU
CPU: 模型部署的主力其实是CPU在做“调度”,纯用CPU做推理当然也行,但很慢。我记得有次用纯CPU跑7B模型,一次推理花了快半分钟……对,你没听错。
- 推理场景:建议8核以上,主频3.0+ GHz。
- 训练场景:核心数可以多,但单核性能也得够。
GPU: 这块水最深。不是所有模型都吃显存,但所有模型都吃“并行度”。
- 推理:6B以下模型,RTX 3060 12G就够了。我劝你一句,显存优先于算力。
- 训练:70B以上大模型,A100 80G起步,否则你连模型都装不进去。
NPU / TPU: 如果你做专用场景,比如AI盒子、边缘端,NPU能效比高很多。但通用性差,我就踩过坑——换了芯片,模型得重新量化适配。
内存与存储:别当短板
内存这块,许多人只看“大不大”,其实“快不快”更关键。我建议:
- 推理:16GB RAM起步,32GB更从容。如果CPU加载模型权重,RAM得是模型大小的2~3倍。
- 训练:64GB +,甚至更高,尤其是你跑数据预处理的时候。
存储,我只有一个建议:NVMe SSD,别用机械盘。你想想看,加载1GB权重到显存,如果是NVMe SSD,几秒搞定;如果是SATA SSD,你等着喝杯咖啡吧。
核心记忆口诀:
模型大小 决定显存下限
推理速度 决定GPU算力
训练时长 决定散热和能力
存储读写 决定不卡不崩
常见配置推荐速查
| 场景 | 典型配置 | 功耗 | 算力参考 | 编程框架 |
|---|---|---|---|---|
| 7B以下模型推理 | NVIDIA Jetson Orin / RTX 3060 12G | 130W ~ 250W | ~13 TFLOPS FP16 | TensorRT / ONNX Runtime |
| 13B ~ 70B模型推理 | RTX 4090 24G / A6000 48G | 250W ~ 300W | ~82 TFLOPS FP16 | vLLM / llama.cpp / TRT |
| 70B+训练 / 微调 | A100 80G / H100 / 4090集群 | 400W ~ 700W | ~312 TFLOPS FP16 | DeepSpeed / Megatron / PyTorch |
| 边缘端 / 离线推理 | Intel x86 + Intel Arc / Apple M2 Pro | 45W ~ 100W | ~5 ~ 10 TFLOPS | OpenVINO / CoreML / llama.cpp |
我个人习惯先查一遍模型官方推荐的硬件配置。比如Meta对Llama 3的要求,比一般开源模型更挑显存。没事别硬上,硬上就是卡死。
异构计算考量:CPU + GPU + NPU 如何协作
哦,这块我得多说几句。异构计算说白了就是让不同硬件干自己擅长的事。
- 预处理:CPU + 数据加载+图像解码,丢给GPU去算。
- 推理核心:GPU做矩阵乘加,NPU做量化加速。
- 后处理:CPU再做解码、NMS、输出组装。
这样一来,GPU不会被琐事拖慢,CPU也不至于闲着。
注意:异构计算不是把模型拆开乱扔。你得做显存调度、通信带宽规划。我曾经在PCIe 3.0上挂两颗GPU,结果互相传数据的时间比算的时间还长。后来换到PCIe 4.0才好。
我建议你在部署前先用工具看看带宽:
nvidia-smi topo -m # 查看GPU拓扑与互联带宽
lscpu --all --extended # 查看CPU NUMA节点
SVG:硬件选型决策流程图
避坑指南:我曾经踩过的几个坑
- 显存不够? 别只想着换卡,先试试量化(INT8/FP16),能省一大半显存。
- 功耗爆炸? 我遇到过一台单卡满载350W,散热顶不住直接降频。后来加了水冷才稳定。算力再强,温度一高,全是白费。
- 异构通信慢? 上面说了,PCIe 3.0 对多GPU通信就是噩梦。建议至少PCIe 4.0,或者考虑NVLink。
- 存储选错? 别用普通SATA SSD做大模型加载盘,你会等到崩溃。买个NVMe吧,真的值得。
总结一句话:
选硬件,先定模型,再定场景,最后选芯片。
算力可以慢慢加,显存和带宽一旦漏了,就得重来。