
直接套现成封装工具
最常见的就是Ollama、LM Studio这类封装好的工具,不用自己处理依赖,上手最快。
- Linux或者WSL用户装Ollama不用找安装包,直接在终端跑一句安装脚本就行:
curl -fsSL https://ollama.com/install.sh | sh - 装完不需要额外配环境变量,直接用ollama run命令拉取对应模型就能对话,想改上下文窗口的话可以编辑用户目录下.ollama文件夹里的配置文件,默认的上下文长度应付日常闲聊、查命令足够。
- 偏好图形界面的可以选LM Studio,搜对应模型点下载就能启动本地服务,默认开放1234端口,完全兼容OpenAI的接口格式,接第三方客户端不用改代码。
这种方式的好处是零配置,模型库都是做过适配优化的,不用自己纠结量化格式;缺点是自定义程度不高,想插自己训的LoRA、改推理逻辑基本做不到,适合只想拿来用、不想花时间折腾环境的人。
用原生推理框架搭链路
llama.cpp、vLLM这类轻量推理框架,自由度比封装工具高不少。
- 先拉取官方仓库代码,提前配好自己设备的GPU加速环境,苹果芯片用户编译的时候记得加LLAMA_METAL=1的参数,N卡用户对应开启CUDA支持,不然纯CPU跑速度会慢不少。
- 自己去HuggingFace下载对应量化精度的GGUF格式模型,放到项目的models目录下,启动服务的时候指定模型路径、上下文长度、GPU卸载层数就行,举个最常用的启动命令例子:
./main -m models/qwen2-7b-instruct-q4_k_m.gguf -ngl 99 -c 8192 - 要做二次开发的话可以直接调server模式的接口,温度、重复惩罚这类生成参数都可以按需自定义。
这种方式的好处是优化空间大,可以根据自己的硬件调到最适配的状态,也支持加自定义插件;缺点是有一定门槛,碰到编译报错、依赖冲突得自己排查,适合有一点基础、想自定义功能的玩家。
基于Python生态直接加载模型
也就是直接用HuggingFace的Transformers库加载模型,做开发最顺手。
- 先建个独立的Python虚拟环境,装好torch、transformers、accelerate这几个核心依赖,注意torch版本要和自己设备的CUDA版本对应,不然会识别不到GPU。
- 加载模型的时候直接传量化参数就行,不用提前下载量化好的模型文件,最简的加载代码也就几行:
from transformers import AutoTokenizer, AutoModelForCausalLM model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-1.5B-Instruct", load_in_4bit=True) tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-1.5B-Instruct") - 后续做微调、挂载LoRA适配器、加RAG链路都可以直接在这个环境里写,不用转模型格式。
这种方式的好处是和整个开源AI生态完全打通,做二次开发、练手项目最方便;缺点是推理速度大概比前两种方案慢一些,依赖包占的存储空间也更大,适合需要做模型微调、定制开发的开发者。
别上来就硬拉7B以上参数的模型,4G内存的老设备先从0.5B、1.5B的q4量化版本试,跑不起来再往下减参数,别跟自己的硬件较劲。
反正这几种方式我平时换着用,写脚本查个命令就用Ollama,要加自定义小功能就切llama.cpp,想喂点自己的数据做微调就开Python环境,根本不存在什么万能方案,适配自己当下的需求就好。
评论一下?