INSTALL.md 1.73 KB
Newer Older
1
# Installation
Shaoshuai Shi's avatar
Shaoshuai Shi committed
2
3
4

### Requirements
All the codes are tested in the following environment:
5
* Linux (tested on Ubuntu 14.04/16.04/18.04/20.04/21.04)
Shaoshuai Shi's avatar
Shaoshuai Shi committed
6
* Python 3.6+
7
* PyTorch 1.1 or higher (tested on PyTorch 1.1, 1,3, 1,5~1.10)
Shaoshuai Shi's avatar
Shaoshuai Shi committed
8
* CUDA 9.0 or higher (PyTorch 1.3+ needs CUDA 9.2+)
9
* [`spconv v1.0 (commit 8da6f96)`](https://github.com/traveller59/spconv/tree/8da6f967fb9a054d8870c3515b1b44eca2103634) or [`spconv v1.2`](https://github.com/traveller59/spconv) or [`spconv v2.x`](https://github.com/traveller59/spconv)
Shaoshuai Shi's avatar
Shaoshuai Shi committed
10
11


12
13
### Install `pcdet v0.5`
NOTE: Please re-install `pcdet v0.5` by running `python setup.py develop` even if you have already installed previous version.
14

Shaoshuai Shi's avatar
Shaoshuai Shi committed
15
16
a. Clone this repository.
```shell
Shaoshuai Shi's avatar
Shaoshuai Shi committed
17
git clone https://github.com/open-mmlab/OpenPCDet.git
Shaoshuai Shi's avatar
Shaoshuai Shi committed
18
19
20
21
```

b. Install the dependent libraries as follows:

22
23
24
25
26
27
28
[comment]: <> (* Install the dependent python libraries: )

[comment]: <> (```)

[comment]: <> (pip install -r requirements.txt )

[comment]: <> (```)
Shaoshuai Shi's avatar
Shaoshuai Shi committed
29

Shaoshuai Shi's avatar
Shaoshuai Shi committed
30
31
32
* Install the SparseConv library, we use the implementation from [`[spconv]`](https://github.com/traveller59/spconv). 
    * If you use PyTorch 1.1, then make sure you install the `spconv v1.0` with ([commit 8da6f96](https://github.com/traveller59/spconv/tree/8da6f967fb9a054d8870c3515b1b44eca2103634)) instead of the latest one.
    * If you use PyTorch 1.3+, then you need to install the `spconv v1.2`. As mentioned by the author of [`spconv`](https://github.com/traveller59/spconv), you need to use their docker if you use PyTorch 1.4+. 
33
34
35
    * You could also install latest `spconv v2.x` with pip, see the official documents of [spconv](https://github.com/traveller59/spconv).
  
c. Install this `pcdet` library and its dependent libraries by running the following command:
Shaoshuai Shi's avatar
Shaoshuai Shi committed
36
37
38
```shell
python setup.py develop
```