1 Star 1 Fork 0

OpenXRLab / xrmogen

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
installation.md 4.19 KB
一键复制 编辑 原始数据 按行查看 历史
gaoyang 提交于 2022-09-01 12:33 . pass pre-commit locally

安装

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

准备环境

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
1
https://gitee.com/OpenXRLab/xrmogen.git
git@gitee.com:OpenXRLab/xrmogen.git
OpenXRLab
xrmogen
xrmogen
main

搜索帮助