GeneFace++(GeneFace Plus Plus) 官方实现主要基于 PyTorch,对显卡的支持情况如下:
1. 官方明确支持的卡
- NVIDIA GPU(CUDA):✅ 主力支持
- 需要 NVIDIA 显卡 + CUDA + cuDNN
- 官方代码、预训练模型、训练/推理脚本都是按 CUDA 环境写的
- 常见可用卡:
- RTX 20 系 / 30 系 / 40 系(2060、3070、3090、4080、4090 等)
- Titan / A100 / V100 等数据中心卡也可
- 显存建议:
- 推理:≥ 6–8 GB
- 训练:≥ 16–24 GB 更稳
2. 其他卡的情况
- AMD GPU(ROCm):❌ 不官方支持
- PyTorch ROCm 理论能跑,但 GeneFace++ 没做适配,容易踩坑
- Intel GPU(Arc / 核显):❌ 不支持
- Apple M 系列(M1/M2/M3):⚠️ 只能 CPU 跑
- 可以跑推理,但非常慢,不支持 Metal 加速训练
- 纯 CPU:✅ 能跑,但很慢
- 仅适合测试,不适合实际用
3. 简单总结
GeneFace++ 基本只推荐用 NVIDIA 显卡(CUDA)
没有 N 卡的话,基本没法愉快用。
如果你愿意,我可以帮你:
- 看你的显卡能不能跑
- 给你最低显存配置建议
- 或推荐替代方案(比如 SadTalker / Wav2Lip)