INSTALLATION.md

May 3, 2020 ยท View on GitHub

Installation

Requirements

  • Linux
  • Python 3.6+
  • PyTorch 1.1 or higher
  • CUDA 10.0 or higher
  • CMake 3.13.2 or higher
  • APEX
  • spconv (A variation of original version)
  • nuscenes-devkit

spconv and nuscenes-devkit should be the specific version from link above

we have tested the following versions of OS and softwares:

  • OS: Ubuntu 16.04/18.04
  • Python: 3.6.5
  • PyTorch: 1.1
  • CUDA: 10.0
  • CUDNN: 7.5.0

Install Requirements

Installation of APEX and spconv should be unver the gpu environment.

APEX

$ git clone https://github.com/NVIDIA/apex
$ cd apex
$ pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

spconv

 $ sudo apt-get install libboost-all-dev
 $ git clone https://github.com/Benzlxs/spconv --recursive
 $ cd spconv && git checkout 7342772
 $ python setup.py bdist_wheel
 $ cd ./dist && pip install *

nuscenes-devkit

$ git clone https://github.com/poodarchu/nuscenes.git
$ cd nuscenes
$ python setup.py install

Install Det3D

The installation should be under the gpu environment.

Clone the Det3D repository

$ git clone https://github.com/poodarchu/Det3D.git
$ cd Det3D

Install Det3D

$ python setup.py build develop

Common Installation Issues

ModuleNotFoundError: No module named 'det3d.ops.nms.nms' when installing det3d

Run python setup.py build develop again.

"values of 'package_data' dict" must be a list of strings (got '*.json') when installing nuscenes-devikit

Use setuptools 39.1.0