install_guide-zh.md

环境配置

English Doc

本文档陈述了搭建MimicTalk Python环境的步骤，我们使用了Conda来管理依赖（与Real3D-Portrait的依赖一致）。

以下配置已在 A100/V100 + CUDA12.1 中进行了验证。

安装Python依赖与CUDA

cd <MimicTalkRoot>
source <CondaRoot>/bin/activate
conda create -n mimictalk python=3.9
conda activate mimictalk

# MMCV for SegFormer network structure
# 其他依赖项
pip install -r docs/prepare_env/requirements.txt -v
pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu121
pip install cython
pip install openmim==0.3.9
mim install mmcv==2.1.0 # 使用mim来加速mmcv安装
## 从源代码build pytorch3d
## 这可能会花费较长时间（可能数十分钟左右）；由于要连接Github，可能经常面临time-out问题，请考虑使用代理。
# 安装pytorch3d之前, 需要安装CUDA-12.1 (https://developer.nvidia.com/cuda-toolkit-archive) 并确保 /usr/local/cuda 指向了 `cuda-12.1` 目录
pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"