All the codes are tested in the following environment:
- Python 3.7+
- PyTorch 1.8
- CUDA 11.1
- spconv v2.x
a. Clone this repository.
git clone https://github.com/CVMI-Lab/PLA.git
b. Install the dependent libraries as follows:
-
Install the dependent Python libraries (Please note that you need to install the correct version of
torch
andspconv
according to your CUDA version):pip install -r requirements.txt
-
Install SoftGroup following its official guidance.
cd pcseg/external_libs/softgroup_ops python3 setup.py build_ext develop cd ../../..
-
Install pcseg
python3 setup.py develop