This guide is about building a python environment for Real3D-Portrait with Conda.
The following installation process is verified in A100/V100 + CUDA11.7.
We recommend to install CUDA 11.7 (which is verified in various types of GPUs), but other CUDA versions (such as 10.2, 12.x) may also work well.
cd <Real3DPortraitRoot>
source <CondaRoot>/bin/activate
conda create -n real3dportrait python=3.9
conda activate real3dportrait
conda install conda-forge::ffmpeg # ffmpeg with libx264 codec to turn images to video
### We recommend torch2.0.1+cuda11.7.
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
# Build from source, it may take a long time (Proxy is recommended if encountering the time-out problem)
pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"
# MMCV for some network structure
pip install cython
pip install openmim==0.3.9
mim install mmcv==2.1.0 # use mim to speed up installation for mmcv
# other dependencies
pip install -r docs/prepare_env/requirements.txt -v