如何配置GeneFace Plus Plus环境

GeneFace++（GeneFace Plus Plus）是一个基于深度学习的语音驱动人脸生成/说话人头生成项目，环境配置相对复杂，涉及 Python、PyTorch、CUDA、音频处理、3D人脸相关库 等。下面我按推荐流程 + 常见坑给你一份可操作的配置指南。

一、GeneFace++ 环境概览

GeneFace++ 通常依赖：

• Python 3.8 / 3.9
• PyTorch 1.12–2.0（需匹配 CUDA）
• CUDA 11.3 / 11.6 / 11.7
• ffmpeg
• PyTorch3D
• librosa、opencv、face-alignment、dlib
• NVIDIA GPU（推荐 ≥ 24GB 显存）

⚠️ 官方仓库（如 SVC-Develop-Team/GeneFacePlusPlus）可能会有细微差异，建议以你使用的 repo 为准。

二、推荐环境（最稳组合）

✅ 最推荐配置（实测可用）

三、环境配置步骤（详细）

1️⃣ 创建 Conda 环境

conda create -n geneface python=3.9 -y
conda activate geneface

2️⃣ 安装 PyTorch（CUDA 11.7）

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 \
torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

验证：

python -c "import torch; print(torch.cuda.is_available())"

3️⃣ 安装 PyTorch3D（⚠️ 易踩坑）

方式一（推荐）：从官方 wheel 安装

pip install fvcore iopath
pip install --no-index --find-links https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py39_cu117_pyt1131/ pytorch3d

如果失败，可尝试源码安装（较复杂）。

4️⃣ 安装 GeneFace++ 项目依赖

进入项目目录：

git clone https://github.com/SVC-Develop-Team/GeneFacePlusPlus.git
cd GeneFacePlusPlus

安装依赖：

pip install -r requirements.txt

⚠️ 常见注意点：

• 若 dlib 安装失败：

conda install -c conda-forge dlib

• 若 face-alignment 报错：

pip install face-alignment==1.3.5

5️⃣ 安装 ffmpeg

sudo apt update
sudo apt install ffmpeg

验证：

ffmpeg -version

6️⃣ 安装 CUDA 扩展（如有）

部分模块需要编译 CUDA 扩展：

cd models/modules
python setup.py build_ext --inplace

若报错，通常是：

• CUDA 版本不匹配
• gcc 版本过高（建议 gcc ≤ 11）

四、数据 & 模型准备

GeneFace++ 通常需要：

• 音频特征（mel、hubert、wav2vec）
• 3DMM 参数
• 预训练模型

常见目录结构：

checkpoints/
└── geneface++
data/
├── audio/
├── video/
└── processed/

下载预训练模型（参考 repo README）：

bash scripts/download_pretrained_models.sh

五、运行 Demo（验证环境）

python inference.py \
--audio_path test.wav \
--model_path checkpoints/geneface++

✅ 成功生成视频即说明环境配置正确。

六、常见报错 & 解决方案

❌ 1. PyTorch3D 安装失败

✅ 解决：

• 严格匹配 Python / CUDA / PyTorch 版本
• 优先使用官方 wheel

❌ 2. CUDA out of memory

✅ 解决：

• 降低 batch_size
• 使用 --crop_size 256
• 推理时加 --no_3dmm

❌ 3. dlib / face-alignment 报错

✅ 解决：

conda install -c conda-forge dlib
pip install face-alignment==1.3.5

七、Docker（可选，最省事）

如果你熟悉 Docker，可以直接用：

docker run --gpus all -it \
-v $(pwd):/workspace \
pytorch/pytorch:1.13.1-cuda11.7-cudnn8-runtime \
bash

再手动安装其余依赖。

八、我可以继续帮你什么？

你可以直接告诉我：
1️⃣ 你的 系统 + GPU 型号
2️⃣ 用的是 哪个 GeneFace++ 仓库
3️⃣ 报错日志（完整 traceback）
我可以 按你的机器精确给你一套配置命令，甚至帮你改 requirements.txt 或 setup.py。