Windows快速安装GeneFace++
一 准备与检查
- 操作系统:Windows 10/11 64位(兼容性更好)。
- Python:建议 Python 3.8(使用 Anaconda 管理环境更稳)。
- 构建工具:安装 Visual Studio Build Tools 2019/2022(勾选“C++ 桌面开发”组件),用于编译可能的 C/C++ 扩展。
- GPU 加速(可选):准备 NVIDIA 显卡驱动、CUDA Toolkit、cuDNN,并确保与后续安装的 PyTorch 版本匹配。
- 硬件建议:CPU i5 及以上、内存 ≥8GB、可用磁盘 ≥10GB。
二 一键式安装流程
- 克隆代码并进入目录
git clone https://github.com/your-repo/GeneFace++.git
cd GeneFace++
说明:将仓库地址替换为实际项目地址。
- 创建并激活虚拟环境(推荐)
conda create -n geneface python=3.8 -y
conda activate geneface
- 安装 Python 依赖
pip install -r requirements.txt
如未提供 requirements.txt,可先安装核心依赖:
pip install numpy opencv-python torch torchvision
- 编译 C/C++ 扩展(如有)
mkdir build && cd build
cmake ..
cmake --build . --config Release
- 安装 Python 包(如有 setup.py)
pip install .
- 快速验证
python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
三 GPU环境配置要点
- 先安装与显卡驱动匹配的 CUDA/cuDNN,再安装与之匹配的 PyTorch(建议通过官方命令或预编译 wheel 安装,避免版本冲突)。
- 安装完成后用 python -c "import torch; print(torch.cuda.is_available())" 验证 GPU 是否可用。
- 若使用 conda,可通过 conda 安装 cudatoolkit(例如:conda install cudatoolkit=11.8 -c nvidia),再用 pip 安装对应版本的 PyTorch,以减少环境冲突。
四 常见问题与快速排查
- 构建失败提示缺少编译工具:安装或修复 Visual Studio Build Tools,确保勾选 C++ 桌面开发;必要时以“x64 Native Tools Command Prompt”运行 cmake 与构建命令。
- 下载依赖慢或失败:更换 pip 源(如 -i https://pypi.tuna.tsinghua.edu.cn/simple),或使用 conda 安装部分依赖。
- 运行示例报缺少模块:确认已激活正确的虚拟环境,并在项目根目录或 examples 目录执行脚本。
- GPU 不可用:核对 CUDA/cuDNN/驱动/PyTorch 版本一致性,重启终端或系统后复测。