GeneFace++ 强烈依赖 GPU + CUDA,CPU 基本跑不了。
sudo yum update -y
sudo yum install -y \
git gcc gcc-c++ make \
ffmpeg ffmpeg-devel \
libglvnd-glx mesa-libGL \
python39 python39-devel创建软链接(可选):
sudo alternatives --set python3 /usr/bin/python3.9wget 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 ~/.bashrcgit clone https://github.com/liuvv/GeneFacePlusPlus.git
cd GeneFacePlusPluspython3.9 -m venv venv
source venv/bin/activate
pip install --upgrade pippip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118验证:
python -c "import torch; print(torch.cuda.is_available())"Truepip install -r requirements.txt⚠️ 常见问题:
opencv-python 在 CentOS 可能缺 sopip install opencv-python-headlesspytorch3d 难装(如果不需要可注释)bash scripts/download_pretrained.sh如果网络慢,可手动下载并放到:
checkpoints/你需要:
目录示例:
data/
audio.wav
ref_video.mp4python inference.py \
--workspace workspace \
--audio data/audio.wav \
--ref_video data/ref_video.mp4输出一般在:
workspace/results/sudo yum install -y ffmpeg ffmpeg-devel --enablerepo=rpmfusion--smallnvidia-smi
python -c "import torch; print(torch.cuda.is_available())"
bash scripts/download_pretrained.sh
python inference.py --audio demo.wav如果你愿意,我可以:
直接把你的 CentOS 版本 + 报错信息 发我即可。