File size: 1,859 Bytes
e34aada |
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 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
# Prepare the Environment
[中文文档](./install_guide-zh.md)
This guide is about building a python environment for Real3D-Portrait with Conda.
The following installation process is verified in A100/V100 + CUDA11.7.
# 1. Install CUDA
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.
# 2. Install Python Packages
```
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
# Install from pytorch3d from conda (For fast installation, Linux only)
conda install pytorch3d::pytorch3d
## Alternatively, a choice of compatibility, build from Github's source code.
## It may take a long time (maybe tens of minutes), 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
If you encounter the following error, please try to install the dependencies with the following command:
pip install -r docs/prepare_env/requirements.txt -v --use-deprecated=legacy-resolver
> ERROR: pip's dependency resolver does not currently take into account all the packages
> that are installed. This behaviour is the source of the following dependency conflicts.
> openxlab 0.0.34 requires setuptools~=60.2.0, but you have setuptools 69.1.1 which is incompatible.
```
|