github.com·2026/10/08·DaoyuanLi2816
canifinetune 为单个消费级NVIDIA GPU微调开源LLM做显存预检,核心估算无需PyTorch、不下载权重。其无torch的TrainingConfig统一配置,CLI、本地demo和推荐共用同一估算器,分解权重、梯度、优化器状态、激活、logits/loss和开销,元数据来自本地/缓存Hub配置而非权重,生成Transformers/PEFT配方并保存模型或adapter,基准覆盖加载与更新。核心pip安装,训练扩展需限定依赖,demo仅绑定loopback。示例Qwen2.5-1.5B-Instruct QLoRA在16 GiB预算下为8.420 GiB并判YES;作者报告在Windows RTX 4080上四个Qwen2.5-0.5B-Instruct LoRA/QLoRA案例预留峰值最高2.398 GiB,新旧预测相同、46.2% MAPE且全部高估,不建立跨GPU精度或OOM概率。兼容仅覆盖原生Windows CPU/CUDA和Linux CPU,不支持WSL GPU、Flash Attention、Liger、预量化基座、远程代码、分布式训练;社区测量opt-in、脱敏且未审核不纳入拟合。
为什么重要 单卡消费级 GPU 微调前先判断显存是否够,核心估算免装 PyTorch、免下权重,省掉一轮试错适合 在单卡消费级 GPU 上做 LoRA/QLoRA 微调的工程师
$ python -m pip install canifinetune==0.4.1
$ canifinetune estimate --model Qwen/Qwen2.5-1.5B-Instruct --method qlora --gpu-vram-gb 16 --seq-len 2048 --offline
$ canifinetune demo
$ pip install "canifinetune[train]==0.4.1"
EVIDENCE / METHODS
要点
单卡消费级 NVIDIA GPU 上训练一个开放权重模型要多少显存,先算再说