安装GeneFace++在Linux的流程

AI技术
小华
2025-12-28

Linux下安装GeneFace++的标准流程
一 环境准备与驱动

  • 操作系统建议:Ubuntu 22.04/24.04(其他发行版亦可,但包管理与驱动安装命令不同)。
  • 显卡与驱动:建议使用NVIDIA显卡,安装与CUDA版本匹配的驱动;例如CUDA 11.7通常搭配驱动版本≥515
  • 安装驱动示例(Ubuntu):
  • 卸载旧驱动:sudo apt-get --purge remove "*nvidia*"
  • 添加显卡驱动PPA并安装:sudo add-apt-repository ppa:graphics-drivers/ppa && sudo apt-get update
  • 安装驱动:sudo apt-get install nvidia-driver-535
  • 安装CUDA Toolkit(示例为11.7):
  • 下载并运行安装器:wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
  • 执行:sudo sh cuda_11.7.0_515.43.04_linux.run(仅勾选安装CUDA Toolkit
  • 配置环境变量(写入~/.bashrc):
  • export PATH="/usr/local/cuda-11.7/bin:$PATH"
  • export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.7/lib64
  • export CUDA_HOME=/usr/local/cuda
  • 验证:nvcc --version 应显示11.7nvidia-smi 可查看驱动与GPU信息。

二 创建Python环境并安装PyTorch与多媒体依赖

  • 建议使用conda隔离环境:
  • 创建环境:conda create -n geneface python=3.9
  • 激活环境:conda activate geneface
  • 安装PyTorch(与CUDA 11.7匹配):
  • conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c nvidia -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch -c https://mirrors.tuna.tsinghua.edu.org/anaconda/cloud/main -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/free
  • 注意:torch 2.1 + CUDA 12.1在部分子模块(如torch-ngp)上可能出现兼容性问题,优先使用2.0.1 + 11.7
  • 多媒体与系统依赖:
  • conda install -c conda-forge ffmpeg(如系统已装可跳过)
  • Ubuntu:sudo apt-get install libasound2-dev portaudio19-dev
  • 其他Python依赖:按项目文档执行,例如:pip install -r docs/prepare_env/requirements.txt -v -i https://pypi.tuna.tsinghua.edu.cn/simple

三 安装3D与可微渲染相关依赖

  • 安装pytorch3d(从源码稳定分支):
  • pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"
  • 安装mmcv(使用mim以加速与匹配版本):
  • pip install cython -i https://pypi.tuna.tsinghua.edu.cn/simple
  • pip install openmim==0.3.9 -i https://pypi.tuna.tsinghua.edu.cn/simple
  • mim install mmcv==2.1.0
  • 构建torch-ngp(项目所需的可微渲染/NeRF组件):
  • 在项目根目录执行构建脚本:bash do(具体脚本名与参数以项目实际README为准)

四 获取GeneFace++源码与项目安装

  • 克隆项目代码(请使用项目官方仓库地址替换示例URL):
  • git clone https://github.com/your-org/GeneFacePlusPlus.git
  • cd GeneFacePlusPlus
  • 安装Python依赖与项目包:
  • pip install -r requirements.txt
  • 可选:pip install -e .(可编辑模式,便于二次开发)
  • 数据准备与目录结构:按项目文档准备训练/测试数据(如视频、音频、配置等),目录结构需与配置保持一致。
  • 小提示:若遇到网络不稳定导致依赖下载缓慢或超时,可使用国内镜像源(如清华源)或在pip命令中添加-i https://pypi.tuna.tsinghua.edu.cn/simple

五 快速验证与常见问题

  • 快速验证:
  • 运行项目提供的测试脚本/示例(如bash scripts/demo.shpython demo.py,以实际文档为准),检查是否能正常加载模型并生成输出视频。
  • 常见问题与排查:
  • 版本不匹配:CUDA 11.7 + torch 2.0.1是较稳的组合;torch 2.1 + CUDA 12.1可能导致torch-ngp报错。
  • 驱动与CUDA:确保nvidia-sminvcc --version显示的驱动与CUDA版本匹配;必要时重设PATHLD_LIBRARY_PATH
  • 构建失败:确认已安装gcc/g++/cmake,并满足项目的编译需求;清理后重试。
  • 依赖冲突:优先在全新conda环境中安装,避免与系统或其他环境包冲突。
  • 运行报错:查看报错栈定位是Python依赖、CUDA算子还是数据路径问题,逐项核对文档与配置。
亿速云提供售前/售后服务

售前业务咨询

售后技术保障

400-100-2938

7*24小时售后电话

官方微信小程序