Installation
December 19, 2022 ยท View on GitHub
System/Hardware Requirements
To get started, the following requirements should be fulfilled.
- System requirements. OpenCOOD is tested under Ubuntu 18.04
- Adequate GPU. A minimum of 6GB gpu is recommended.
- Disk Space. Estimate 100GB of space is recommended for data downoading.
- Python Python3.7 is required.
Installation
1. Dependency Installation
First, download OpenCOOD github to your local folder if you haven't done it yet.
git clone https://github.com/DerrickXuNu/OpenCOOD.git
cd OpenCOOD
Next we create a conda environment and install the requirements.
conda env create -f environment.yml
conda activate opencood
python setup.py develop
If conda install failed, install through pip
pip install -r requirements.txt
2. Pytorch Installation (>=1.8, tested on 1.8-1.12.0)
Go to https://pytorch.org/ to install pytorch cuda version.
3. Spconv (1.2.1 or 2.x)
OpenCOOD support both spconv 1.2.1 and 2.x to generate voxel features.
To install spconv 1.2.1, please follow the guide in https://github.com/traveller59/spconv/tree/v1.2.1.
To install spconv 2.x, please run the following commands (if you are using cuda 11.3):
pip install spconv-cu113
Tips for installing spconv 1.2.1:
- make sure your cmake version >= 3.13.2
- CUDNN and CUDA runtime library (use
nvcc --versionto check) needs to be installed on your machine.
4. Bbx IOU cuda version compile
Install bbx nms calculation cuda version
python opencood/utils/setup.py build_ext --inplace
5. Dependencies for FPV-RCNN (optional)
Install the dependencies for fpv-rcnn.
python opencood/pcdet_utils/setup.py build_ext --inplace