Newer
Older
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
- PyTorch 1.6+
- CUDA 9.2+
- GCC 5.4+
### 准备环境
1. 使用 conda 新建虚拟环境，并进入该虚拟环境；
```bash
conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab
```
2. 安装 PyTorch
在安装 MMEngine 之前，请确保 PyTorch 已经成功安装在环境中，可以参考 [PyTorch 官方安装文档](https://pytorch.org/get-started/locally/#start-locally)。使用以下命令验证 PyTorch 是否安装
```bash
python -c 'import torch;print(torch.__version__)'
```
### 安装 MMEngine
#### 使用 mim 安装
[mim](https://github.com/open-mmlab/mim) 是 OpenMMLab 项目的包管理工具，使用它可以很方便地安装 OpenMMLab 项目。
```bash
pip install -U openmim
mim install mmengine
```
#### 使用 pip 安装
```bash
pip install mmengine
```
#### 使用 docker 镜像
1. 构建镜像
```bash
docker build -t mmengine https://github.com/open-mmlab/mmengine.git#main:docker/release
```
更多构建方式请参考 [mmengine/docker](https://github.com/open-mmlab/mmengine/tree/main/docker)。
2. 运行镜像
```bash
docker run --gpus all --shm-size=8g -it mmengine
```
#### 源码安装
```bash
# 如果克隆代码仓库的速度过慢，可以从 https://gitee.com/open-mmlab/mmengine.git 克隆
git clone https://github.com/open-mmlab/mmengine.git
cd mmengine
pip install -e . -v
```
### 验证安装
为了验证是否正确安装了 MMEngine 和所需的环境，我们可以运行以下命令
```bash
python -c 'import mmengine;print(mmengine.__version__)'
```