"test/srt/vscode:/vscode.git/clone" did not exist on "308d024092d8a671998b978f419dd40262bef9b5"
install.md 4.75 KB
Newer Older
zhangwenwei's avatar
zhangwenwei committed
1
2
3
4
## Installation

### Requirements

zhangwenwei's avatar
Doc  
zhangwenwei committed
5
6
7
8
- Linux or macOS (Windows is not currently officially supported)
- Python 3.6+
- PyTorch 1.3+
- CUDA 9.2+ (If you build PyTorch from source, CUDA 9.0 is also compatible)
zhangwenwei's avatar
zhangwenwei committed
9
- GCC 5+
zhangwenwei's avatar
zhangwenwei committed
10
11
12
13
14
15
16
17
- [mmcv](https://github.com/open-mmlab/mmcv)


### Install mmdetection

a. Create a conda virtual environment and activate it.

```shell
zhangwenwei's avatar
Doc  
zhangwenwei committed
18
conda create -n open-mmlab python=3.7 -y
zhangwenwei's avatar
zhangwenwei committed
19
20
21
22
23
24
25
26
27
conda activate open-mmlab
```

b. Install PyTorch and torchvision following the [official instructions](https://pytorch.org/), e.g.,

```shell
conda install pytorch torchvision -c pytorch
```

zhangwenwei's avatar
Doc  
zhangwenwei committed
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
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/).

`E.g.1` If you have CUDA 10.1 installed under `/usr/local/cuda` and would like to install
PyTorch 1.5, you need to install the prebuilt PyTorch with CUDA 10.1.

```python
conda install pytorch cudatoolkit=10.1 torchvision -c pytorch
```

`E.g. 2` If you have CUDA 9.2 installed under `/usr/local/cuda` and would like to install
PyTorch 1.3.1., you need to install the prebuilt PyTorch with CUDA 9.2.

```python
conda install pytorch=1.3.1 cudatoolkit=9.2 torchvision=0.4.2 -c pytorch
```

If you build PyTorch from source instead of installing the prebuilt pacakge,
you can use more CUDA versions such as 9.0.

zhangwenwei's avatar
zhangwenwei committed
48
49
50
51
52
53
54
55
c. Clone the mmdetection repository.

```shell
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
```

d. Install build requirements and then install mmdetection.
zhangwenwei's avatar
zhangwenwei committed
56
57
(We install our forked version of pycocotools via the github repo instead of pypi
for better compatibility with our repo.)
zhangwenwei's avatar
zhangwenwei committed
58
59
60

```shell
pip install -r requirements/build.txt
61
pip install cython  # cython is necessary for pycocotools
zhangwenwei's avatar
zhangwenwei committed
62
pip install "git+https://github.com/open-mmlab/cocoapi.git#subdirectory=pycocotools"
zhangwenwei's avatar
zhangwenwei committed
63
64
65
pip install -v -e .  # or "python setup.py develop"
```

zhangwenwei's avatar
Doc  
zhangwenwei committed
66
If you build mmdetection on macOS, replace the last command with
zhangwenwei's avatar
zhangwenwei committed
67
68

```
zhangwenwei's avatar
Doc  
zhangwenwei committed
69
CC=clang CXX=clang++ CFLAGS='-stdlib=libc++' pip install -e .
zhangwenwei's avatar
zhangwenwei committed
70
71
```

72
73
74
75
76
77
78
79
80
81
82
83
84
85
e. Clone the mmdetection3d repository.

```shell
git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d
```

f.Install build requirements and then install mmdetection3d.

```shell
pip install -r requirements/build.txt
pip install -v -e .  # or "python setup.py develop"
```

zhangwenwei's avatar
zhangwenwei committed
86
87
88
89
90
Note:

1. The git commit id will be written to the version number with step d, e.g. 0.6.0+2e7045c. The version will also be saved in trained models.
It is recommended that you run step d each time you pull some updates from github. If C++/CUDA codes are modified, then this step is compulsory.

zhangwenwei's avatar
zhangwenwei committed
91
92
93
    > Important: Be sure to remove the `./build` folder if you reinstall mmdet with a different CUDA/PyTorch version.

    ```
94
    pip uninstall mmdet3d
zhangwenwei's avatar
zhangwenwei committed
95
96
97
98
    rm -rf ./build
    find . -name "*.so" | xargs rm
    ```

zhangwenwei's avatar
zhangwenwei committed
99
100
101
102
103
104
105
2. Following the above instructions, mmdetection is installed on `dev` mode, any local modifications made to the code will take effect without the need to reinstall it (unless you submit some commits and want to update the version number).

3. If you would like to use `opencv-python-headless` instead of `opencv-python`,
you can install it before installing MMCV.

4. Some dependencies are optional. Simply running `pip install -v -e .` will only install the minimum runtime requirements. To use optional dependencies like `albumentations` and `imagecorruptions` either install them manually with `pip install -r requirements/optional.txt` or specify desired extras when calling `pip` (e.g. `pip install -v -e .[optional]`). Valid keys for the extras field are: `all`, `tests`, `build`, and `optional`.

106
5. The code can not be built for CPU only environment (where CUDA isn't available) for now.
zhangwenwei's avatar
Doc  
zhangwenwei committed
107

zhangwenwei's avatar
zhangwenwei committed
108
109
110

### A from-scratch setup script

zhangwenwei's avatar
Doc  
zhangwenwei committed
111
Here is a full script for setting up mmdetection with conda.
zhangwenwei's avatar
zhangwenwei committed
112
113

```shell
zhangwenwei's avatar
Doc  
zhangwenwei committed
114
conda create -n open-mmlab python=3.7 -y
zhangwenwei's avatar
zhangwenwei committed
115
116
conda activate open-mmlab

zhangwenwei's avatar
Doc  
zhangwenwei committed
117
# install latest pytorch prebuilt with the default prebuilt CUDA version (usually the latest)
zhangwenwei's avatar
zhangwenwei committed
118
119
120
121
conda install -c pytorch pytorch torchvision -y
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
122
pip install cython
zhangwenwei's avatar
zhangwenwei committed
123
pip install "git+https://github.com/open-mmlab/cocoapi.git#subdirectory=pycocotools"
zhangwenwei's avatar
zhangwenwei committed
124
pip install -v -e .
125
126
127
128
129
130

cd ..
git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d
pip install -r requirements/build.txt
pip install -v -e .
zhangwenwei's avatar
zhangwenwei committed
131
132
```

133
### Using multiple MMDetection3D versions
zhangwenwei's avatar
zhangwenwei committed
134

135
The train and test scripts already modify the `PYTHONPATH` to ensure the script use the MMDetection3D in the current directory.
zhangwenwei's avatar
zhangwenwei committed
136

137
To use the default MMDetection3D installed in the environment rather than that you are working with, you can remove the following line in those scripts
zhangwenwei's avatar
zhangwenwei committed
138
139

```shell
zhangwenwei's avatar
Doc  
zhangwenwei committed
140
PYTHONPATH="$(dirname $0)/..":$PYTHONPATH
zhangwenwei's avatar
zhangwenwei committed
141
```