All the codes are tested in the following environment:
- Linux (tested on Ubuntu 18.04)
- Python 3.8
- PyTorch 1.7
- CUDA 10.2
spconv v1.2 (commit abf0acf)
NOTE: Please re-install pcdet v0.1
by running python setup.py develop
even if you have already installed previous version.
a. Clone this repository.
git clone https://github.com/TRAILab/PDV.git
b. Install the dependent libraries as follows:
- Install the dependent python libraries:
pip install -r requirements.txt
- Install the SparseConv library, we use the implementation from
[spconv]
.- If you use PyTorch 1.1, then make sure you install the
spconv v1.0
with (commit 8da6f96) 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 ofspconv
, you need to use their docker if you use PyTorch 1.4+.
- If you use PyTorch 1.1, then make sure you install the
c. Install this pcdet
library by running the following command:
python setup.py develop