部署DiffPortrait360需要分步骤完成环境准备:
Preparação do hardware
- 必须使用NVIDIA GPU(支持CUDA)
- 最低配置:30GB显存(生成32帧)
- 推荐配置:80GB显存(如A6000显卡)
instalação do software
- 确认CUDA版本(推荐12.2):通过
nvcc --version
sonda - 创建Python 3.9的Conda环境:
conda create -n diffportrait360 python=3.9
conda activate diffportrait360
- Clonar o repositório de código:
git clone https://github.com/FreedomGu/DiffPortrait360.git
- Instalar dependências:
- entrar em
diffportrait360_release
diretório (no disco rígido do computador) - realizar
pip install -r requirements.txt
- entrar em
Implementação do modelo
从Hugging Face下载三个预训练模型(PANO_HEAD_MODEL/Head_Back_MODEL/Diff360_MODEL)并修改inference.sh
中的路径参数。最后通过torch.cuda.is_available()
验证GPU可用性。
Essa resposta foi extraída do artigoDiffPortrait360: gere visualizações de cabeça em 360 graus a partir de um único retratoO