CentOS怎么配置GeneFace Plus Plus

AI技术
小华
2026-08-02

GeneFace++(GeneFace Plus Plus)是一个基于 NeRF 的说话人头像生成项目,对系统环境、GPU、依赖库要求比较严格。下面我按“推荐环境 → 安装 → 配置 → 运行 → 常见坑”的顺序,给你一个CentOS 上可用的实操级配置指南

⚠️ 先说明:
GeneFace++ 强烈依赖高版本 GCC、CUDA、PyTorch,CentOS 7 默认软件非常老,建议用 CentOS 8 / Rocky / Alma,或者容器 / Conda

一、推荐环境(非常重要)

✅ 系统

组件推荐
OSCentOS 8 / Rocky Linux 8 / AlmaLinux 8
GPUNVIDIA ≥ 3090 / A100(16G+ 显存)
CUDA11.7 / 11.8
Python3.9
如果用 CentOS 7,必须用 Docker / Conda 升级 GCC

二、系统基础环境准备(CentOS)

1️⃣ 安装基础工具

sudo dnf groupinstall "Development Tools" -y
sudo dnf install wget curl git unzip -y

2️⃣ 升级 GCC(CentOS 7 必做)

sudo yum install centos-release-scl -y
sudo yum install devtoolset-11 -y
scl enable devtoolset-11 bash

验证:

gcc --version

三、安装 NVIDIA 驱动 & CUDA

1️⃣ 安装驱动

lspci | grep -i nvidia

去 NVIDIA 官网下载驱动并安装。

2️⃣ 安装 CUDA 11.8

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_linux.run

配置环境变量:

echo 'export PATH=/usr/local/cuda-11.8/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc

四、Python & Conda 环境

1️⃣ 安装 Miniconda

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh

2️⃣ 创建 GeneFace++ 环境

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

五、安装 PyTorch(关键)

pip install torch==2.0.1 torchvision torchaudio \
--index-url https://download.pytorch.org/whl/cu118

验证:

import torch
print(torch.cuda.is_available())

六、克隆 & 安装 GeneFace++

1️⃣ 克隆仓库

git clone https://github.com/yerfor/GeneFacePlusPlus.git
cd GeneFacePlusPlus

2️⃣ 安装依赖

pip install -r requirements.txt

常见额外依赖:

pip install librosa transformers accelerate

七、核心模块配置

1️⃣ 音频特征提取

pip install pyworld torchaudio soundfile

2️⃣ NeRF / 3D 依赖

GeneFace++ 通常依赖:

  • tinycudann
  • nvdiffrast

安装 tinycudann

pip install git+https://github.com/NVlabs/tiny-cuda-nn.git

⚠️ 如果失败,先确认:

which nvcc

八、下载模型权重(必需)

GeneFace++ 需要:

  • 预训练说话人模型
  • 音频编码器权重
  • NeRF checkpoint

示例:

mkdir -p checkpoints

按官方 README 下载并放到:

checkpoints/
├── audio_encoder.pt
├── geneface_nerf.pt

九、运行示例

python inference.py \
--audio_path test.wav \
--model_path checkpoints/geneface_nerf.pt

十、CentOS 常见坑(非常重要)

❌ 1. GCC 版本太低

现象:

error: ‘std::byte’ has not been declared

✅ 解决:

scl enable devtoolset-11 bash

❌ 2. torch.cuda.is_available() = False

原因:

  • CUDA ≠ PyTorch CUDA 版本

✅ 检查:

python -m torch.utils.collect_env

❌ 3. tinycudann 编译失败

原因:

  • 没装 CUDA Toolkit
  • GCC 版本不兼容

✅ 确认:

nvcc --version

十一、我可以进一步帮你

你可以直接把下面信息发我,我给你“量身方案”

  1. ✅ CentOS 版本(7 or 8)
  2. ✅ GPU 型号
  3. ✅ 是否用 Docker
  4. ✅ 报错信息(完整)

我可以直接帮你:

  • ✅ 写 Dockerfile
  • ✅ 写 一键安装脚本
  • ✅ 针对 RTX 3090 / A100 优化配置

想继续吗?

亿速云提供售前/售后服务

售前业务咨询

售后技术保障

400-100-2938

7*24小时售后电话

官方微信小程序