File size: 1,142 Bytes
4abefb9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# 环境配置
[English Doc](./install_guide.md)

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

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


# 安装Python依赖与CUDA
```bash
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"
```