You can either build the docker image through Dockerfile or pull the docker image from dockerhub. Please make sure nvidia-docker is corretly installed.
Build docker image that support OpenPCDet through:
docker build ./ -t openpcdet-docker
Note that if you would like to use dynamic voxelization, you need further install torch_scatter
package.
From this Dockerfile, the installed version of spconv is 2.x, if you would like to use spconv 1.2.1, please follow these steps:
git clone -b v1.2.1 https://github.com/djiajunustc/spconv spconv --recursive
cd spconv
python setup.py bdist_wheel
cd ./dist
pip install *.whl
Run the following script to pull the docker image:
docker pull djiajun1206/pcdet:pytorch1.6