表情迁移项目LivePortrait本地部署中的坑
前提准备:
从Nvidia官网下载最新版的驱动程序:https://www.nvidia.com/en-us/drivers/
授权.run文件的运行权限:
sudo chmod +x filename.run
---------------------------------------------------------------------------------
1、克隆项目至本地:
git clone https://github.com/KwaiVGI/LivePortrait
2、创建虚拟环境:
conda create -n LivePortrait python=3.11 && conda activate LivePortrait
官方文档让创建3.10的python,我改为了3.11
3、安装最新版scipy
pip install scipy -i https://pypi.tuna.tsinghua.edu.cn/simple
4、修改requirements_base.txt文件
- 去掉其中的scipy==1.13.1一行
- 把gradio的版本号去掉(无版本号则安装最新版本)
5、在pytorch官网(https://pytorch.org/get-started/previous-versions/)找到对应(或相近的)cuda版本安装命令,并安装:
我RTX 3060安装后的CUDA是12.8,因此,找到12.6的安装命令:
pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu126
pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu126
6、安装项目依赖库:
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
7、下载模型:
pip install -U huggingface_hub -i https://pypi.tuna.tsinghua.edu.cn/simple
export HF_ENDPOINT=https://hf-mirror.com
huggingface-cli download KwaiVGI/LivePortrait --local-dir pretrained_weights --exclude "*.git*" "README.md" "docs"
8、运行程序
python app.py
或命令行命令
# source input is an image
python inference.py -s assets/examples/source/s9.jpg -d assets/examples/driving/d0.mp4
# source input is a video ✨
python inference.py -s assets/examples/source/s13.mp4 -d assets/examples/driving/d0.mp4
评论
发表评论