installation.md

安装

本文档提供了安装 XRNerf 的相关步骤。

• 安装

  • 准备环境

    • a. 创建并激活 conda 虚拟环境.
    • b. 安装 PyTorch 和 torchvision
    • c. 安装MMHuman3D
    • d. 安装其他python包

  • 利用 Docker 镜像安装 XRMoGen

准备环境

a. 创建并激活 conda 虚拟环境.

conda create -n xrmogen python=3.8 -y
conda activate xrmogen
conda install ffmpeg -y

b. 安装 PyTorch 和 torchvision

1. 查看pytorch-cuda版本匹配表，选择合适的版本 here
2. 用对应conda install 命令安装对应版本的PyTorch以及Torchvision。

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

c. 安装 MMHuman3D

mkdir ./workspace && cd ./workspace
git clone https://github.com/open-mmlab/mmhuman3d.git
cd mmhuman3d && pip install -e .
cd ../..

d. 安装其他python包

pip install -r requirements.txt
pip install -e .

利用 Docker 镜像安装 XRMoGen

XRMogen 提供一个 Dockerfile 可以直接创建 docker 镜像

docker build -f ./Dockerfile --rm -t xrmogen .

注意 用户需要确保已经安装了 nvidia-container-toolkit。

运行以下命令，进入镜像:

docker run --gpus all -it xrmogen /workspace

在本机上(非docker镜像机内)开启一个终端，将项目文件(包括数据集)复制进docker镜像机

docker cp ProjectPath/xrmogen [DOCKER_ID]:/workspace

其中[DOCKER_ID] 是镜像的id, 通过下面命令确定

docker ps -a