GeneFace++(GeneFace Plus Plus) 对显存和显卡的适配情况大致如下(基于官方仓库与社区实践):
GeneFace++ 基于 PyTorch,因此主要适配 NVIDIA CUDA 显卡,部分情况下也可跑在 NVIDIA 数据中心卡 / 消费级卡 上:
⚠️ 不支持:
- AMD 显卡(ROCm 基本不可用)
- 苹果 M 系列(Metal 不支持 CUDA)
- 纯 CPU 推理(极慢,不现实)
| 任务 | 最低显存 | 推荐显存 |
|---|---|---|
| 推理(已训练模型) | 6–8 GB | 12 GB+ |
| 微调 / 训练 | 16 GB | 24 GB+ |
| 全量训练 | 24 GB+ | 40 GB+ |
batch_sizefp16 / bf16gradient_checkpointing最适配:
RTX 3090 / 4090、A100
能跑但受限:
3060 12GB、3080 12GB
如果你愿意,我可以按你的具体显卡型号告诉你能不能跑、怎么配参数。