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

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


Shaoshuai Shi's avatar
Shaoshuai Shi committed
12
### Install `pcdet v0.3`
13
NOTE: Please re-install `pcdet v0.3` 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
22
23
24
25
26
```

b. Install the dependent libraries as follows:

* Install the dependent python libraries: 
```
pip install -r requirements.txt 
```

Shaoshuai Shi's avatar
Shaoshuai Shi committed
27
28
29
* 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+. 
Shaoshuai Shi's avatar
Shaoshuai Shi committed
30
31
32
33
34

c. Install this `pcdet` library by running the following command:
```shell
python setup.py develop
```