CentOS远程安装GeneFace++方法

AI技术
小华
2025-11-26

CentOS远程安装 GeneFace++ 实操指南
一 环境准备与远程连接

  • 建议使用 CentOS 7/8,具备 NVIDIA GPU 并安装对应版本的 CUDA/cuDNN 与驱动(推理/训练通常需要 GPU)。远程操作建议使用 SSH,如需图形界面可用 VNC/X2Go 或端口转发运行 WebUI。
  • 基础工具与开发包(无 GPU 也可先准备 Python 环境,但推理训练需要 GPU):
  • 安装基础工具:sudo yum update -y
  • 安装编译与图像依赖:sudo yum install -y cmake python3 python3-devel libpng-devel libjpeg-devel libtiff-devel
  • 安装开发工具组:sudo yum groupinstall -y "Development Tools"
  • 可选:配置国内 YUM 镜像源 以加速下载。

二 获取源码与创建虚拟环境

  • 克隆官方仓库(请确认实际仓库地址与分支):
  • git clone https://github.com/yerfor/GeneFacePlusPlus.git
  • cd GeneFacePlusPlus
  • 创建并激活 conda 虚拟环境(推荐 Python 3.9):
  • conda create -n geneface python=3.9 -y
  • conda activate geneface
  • 安装 Python 依赖(按项目要求):
  • pip install -r docs/prepare_env/requirements.txt -v
  • 安装 PyTorch 2.0.1 + cu117(示例,按你的 CUDA 版本调整):
  • pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
  • 安装 pytorch3d(稳定版)与 MMCV
  • pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"
  • pip install cython openmim==0.3.9
  • mim install mmcv==2.1.0
  • 音频依赖(若本机为 CentOS 7 且需音频设备/录制,可后续补充 ALSA/PortAudio 开发库)。

三 CUDA 与驱动安装要点(仅 GPU 场景)

  • 确认驱动与 CUDA 匹配,常见组合为 CUDA 11.7 + 驱动 535(示例):
  • 查看驱动:nvidia-smi
  • 若无驱动或版本不符,先安装合适版本的 NVIDIA 驱动,再安装 CUDA 11.7(或使用 conda 的 cudatoolkit 与系统驱动配合)。
  • 注意:CUDA 安装与驱动步骤在 CentOSUbuntu 的包管理命令不同;在 CentOS 上优先使用 NVIDIA 官方 runfile 或 ELRepo 方式,避免与系统自带的 nouveau 冲突。

四 项目构建与数据准备

  • 构建扩展与依赖(项目提供脚本):
  • bash docs/prepare_env/install_ext.sh
  • 准备 3DMM 模型 BFM2009(放置到项目路径):
  • 将 BFM 文件夹(含 01_MorphableModel.mat 等 8 个文件)移动到:deep_3drecon/BFM/
  • 准备示例数据(示例为 May 数据集):
  • 下载 trainval_dataset.npy,放置到:data/binary/videos/May/trainval_dataset.npy
  • 说明:若从源码构建失败,优先检查 gcc/g++/cmake 版本、CUDA 与 PyTorch 版本一致性,以及 mmcv 是否安装成功。

五 推理验证与远程访问

  • 命令行推理示例(确保 CUDA_VISIBLE_DEVICES 指向可用 GPU):
  • export PYTHONPATH=.
  • python inference/genefacepp_infer.py --a2m_ckpt=checkpoints/audio2motion_vae --torso_ckpt=checkpoints/motion2video_nerf/may_torso --drv_aud=data/raw/val_wavs/MacronSpeech.wav --out_name=may_demo.mp4
  • 启动 Gradio WebUI(便于远程访问):
  • python inference/app_genefacepp.py --a2m_ckpt=checkpoints/audio2motion_vae --torso_ckpt=checkpoints/motion2video_nerf/may_torso
  • 远程访问方式:
  • 服务器有公网 IP:直接访问 http://<服务器IP>:<端口>
  • 仅内网:本机端口转发 ssh -L 7860:127.0.0.1:7860 @,然后浏览器访问 http://127.0.0.1:7860
  • 常见问题与优化:
  • 模型从 Hugging Face 下载慢或失败:设置镜像 export HF_ENDPOINT=https://hf-mirror.com
  • 若出现 OpenMP 冲突:可临时设置 export KMP_DUPLICATE_LIB_OK="TRUE"(仅用于排查)
  • 防火墙/云安全组:开放对应端口(如 7860)后再访问。
亿速云提供售前/售后服务

售前业务咨询

售后技术保障

400-100-2938

7*24小时售后电话

官方微信小程序