mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmocr/docs_zh_CN/install.md

152 lines
4.5 KiB
Markdown
Raw Normal View History

[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
# Installation
## Prerequisites
- Linux (Windows is not officially supported)
- Python 3.7
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
- PyTorch 1.6 or higher
- torchvision 0.7.0
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
- CUDA 10.1
- NCCL 2
- GCC 5.4.0 or higher
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
- [MMCV](https://mmcv.readthedocs.io/en/latest/#installation) >= 1.3.8
- [MMDetection](https://mmdetection.readthedocs.io/en/latest/#installation) >= 2.13.0
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
We have tested the following versions of OS and softwares:
- OS: Ubuntu 16.04
- CUDA: 10.1
- GCC(G++): 5.4.0
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
- MMCV 1.3.8
- MMDetection 2.13.0
- PyTorch 1.6.0
- torchvision 0.7.0
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
MMOCR depends on PyTorch and mmdetection.
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
## Step-by-Step Installation Instructions
a. Create a conda virtual environment and activate it.
```shell
conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab
```
b. Install PyTorch and torchvision following the [official instructions](https://pytorch.org/), e.g.,
```shell
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
```
Note: Make sure that your compilation CUDA version and runtime CUDA version match.
You can check the supported CUDA version for precompiled packages on the [PyTorch website](https://pytorch.org/).
c. Install mmcv, we recommend you to install the pre-build mmcv as below.
```shell
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
```
Please replace ``{cu_version}`` and ``{torch_version}`` in the url to your desired one. For example, to install the latest ``mmcv-full`` with ``CUDA 11`` and ``PyTorch 1.7.0``, use the following command:
```shell
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html
```
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
Note that mmocr 0.2.1 or later requires mmcv 1.3.8 or later.
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
If it compiles during installation, then please check that the cuda version and pytorch version **exactly** matches the version in the mmcv-full installation command. For example, pytorch 1.7.0 and 1.7.1 are treated differently.
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
See official [installation](https://github.com/open-mmlab/mmcv#installation) for different versions of MMCV compatible to different PyTorch and CUDA versions.
**Important:** You need to run `pip uninstall mmcv` first if you have mmcv installed. If mmcv and mmcv-full are both installed, there will be `ModuleNotFoundError`.
d. Install [mmdet](https://github.com/open-mmlab/mmdetection.git), we recommend you to install the latest `mmdet` with pip.
See [here](https://pypi.org/project/mmdet/) for different versions of `mmdet`.
```shell
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
pip install mmdet
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
```
Optionally you can choose to install `mmdet` following the official [installation](https://github.com/open-mmlab/mmdetection/blob/master/docs/get_started.md).
e. Clone the mmocr repository.
```shell
git clone https://github.com/open-mmlab/mmocr.git
cd mmocr
```
f. Install build requirements and then install MMOCR.
```shell
pip install -r requirements.txt
pip install -v -e . # or "python setup.py develop"
export PYTHONPATH=$(pwd):$PYTHONPATH
```
## Full Set-up Script
Here is the full script for setting up mmocr with conda.
```shell
conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab
# install latest pytorch prebuilt with the default prebuilt CUDA version (usually the latest)
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
# install the latest mmcv-full
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
# install mmdetection
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
pip install mmdet
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
# install mmocr
git clone https://github.com/open-mmlab/mmocr.git
cd mmocr
pip install -r requirements.txt
pip install -v -e . # or "python setup.py develop"
export PYTHONPATH=$(pwd):$PYTHONPATH
```
## Another option: Docker Image
We provide a [Dockerfile](https://github.com/open-mmlab/mmocr/blob/master/docker/Dockerfile) to build an image.
```shell
[Docs] update dependency version req, dockerfile and change logs for 0.2.1 (#331) * update pytorch req and dockerfile * Update dependency requirement * update readme for 0.2.1 * update change log * update release date
2021-07-20 23:18:47 +08:00
# build an image with PyTorch 1.6, CUDA 10.1
[Docs] Add cn docs framework (#353) * add CN demo docs * add deployment.md to docs * add placeholder CN docs * Add language switching hint
2021-07-07 14:13:27 +08:00
docker build -t mmocr docker/
```
Run it with
```shell
docker run --gpus all --shm-size=8g -it -v {DATA_DIR}:/mmocr/data mmocr
```
## Prepare Datasets
It is recommended to symlink the dataset root to `mmocr/data`. Please refer to [datasets.md](datasets.md) to prepare your datasets.
If your folder structure is different, you may need to change the corresponding paths in config files.
The `mmocr` folder is organized as follows:
```
├── configs/
├── demo/
├── docker/
├── docs/
├── LICENSE
├── mmocr/
├── README.md
├── requirements/
├── requirements.txt
├── resources/
├── setup.cfg
├── setup.py
├── tests/
├── tools/
```