Shortcuts

安装

环境依赖

  • Python 3.7+

  • PyTorch 1.6+

  • CUDA 9.2+

  • GCC 5.4+

准备环境

  1. 使用 conda 新建虚拟环境,并进入该虚拟环境;

    conda create -n open-mmlab python=3.7 -y
    conda activate open-mmlab
    
  2. 安装 PyTorch

    在安装 MMEngine 之前,请确保 PyTorch 已经成功安装在环境中,可以参考 PyTorch 官方安装文档。使用以下命令验证 PyTorch 是否安装

    python -c 'import torch;print(torch.__version__)'
    

安装 MMEngine

使用 mim 安装

mim 是 OpenMMLab 项目的包管理工具,使用它可以很方便地安装 OpenMMLab 项目。

pip install -U openmim
mim install mmengine

使用 pip 安装

pip install mmengine

使用 docker 镜像

  1. 构建镜像

    docker build -t mmengine https://github.com/open-mmlab/mmengine.git#main:docker/release
    

    更多构建方式请参考 mmengine/docker

  2. 运行镜像

    docker run --gpus all --shm-size=8g -it mmengine
    

源码安装

# 如果克隆代码仓库的速度过慢,可以从 https://gitee.com/open-mmlab/mmengine.git 克隆
git clone https://github.com/open-mmlab/mmengine.git
cd mmengine
pip install -e . -v

验证安装

为了验证是否正确安装了 MMEngine 和所需的环境,我们可以运行以下命令

python -c 'import mmengine;print(mmengine.__version__)'
Read the Docs v: latest
Versions
latest
stable
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.