update readme

This commit is contained in:
lihengzhong 2024-01-06 21:17:34 +08:00
parent 031ef30193
commit 007601704d
1 changed files with 4 additions and 2 deletions

View File

@ -4,12 +4,14 @@ A streaming digital human based on the Ernerf model realize audio video synch
## Installation
Tested on Ubuntu 18.04, Pytorch 1.12 and CUDA 11.3.
Tested on Ubuntu 18.04, Python3.10, Pytorch 1.12 and CUDA 11.3
### Install dependency
```bash
pip install pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.3 -c pytorch
conda create -n nerfstream python=3.10
conda install pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.3 -c pytorch
conda activate nerfstream
pip install -r requirements.txt
pip install "git+https://github.com/facebookresearch/pytorch3d.git"
pip install tensorflow-gpu==2.8.0