- Nvidia driver 440.82
- CUDA 10.2
- pytorch 1.4.0
- python 3.7.6
- Please refer to INSTALL.md for installation and dataset preparation.
- Install detectron 2
# for CUDA 10.2:
python -m pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/index.html
Or refer to Detectron 2 3. Minkowski Engine
./run.sh