Skip to content

Latest commit

 

History

History
25 lines (21 loc) · 911 Bytes

README.md

File metadata and controls

25 lines (21 loc) · 911 Bytes
Raw

Guidance to use OpenPCDet with docker

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 Through Dockerfile

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

Pull From Dockerhub

Run the following script to pull the docker image:

docker pull djiajun1206/pcdet:pytorch1.6