mirror of
https://github.com/botastic/SoftGroup.git
synced 2025-10-16 11:45:42 +00:00
53 lines
913 B
Markdown
53 lines
913 B
Markdown
## Installation
|
|
|
|
1\) Environment requirements
|
|
|
|
* Python 3.x
|
|
* Pytorch 1.1 or higher
|
|
* CUDA 9.2 or higher
|
|
* gcc-5.4 or higher
|
|
|
|
Create a conda virtual environment and activate it.
|
|
```
|
|
conda create -n softgroup python=3.7
|
|
conda activate softgroup
|
|
```
|
|
|
|
|
|
2\) Clone the repository.
|
|
```
|
|
git clone https://github.com/thangvubk/SoftGroup.git --recursive
|
|
```
|
|
|
|
|
|
3\) Install the requirements.
|
|
```
|
|
cd SoftGroup
|
|
pip install -r requirements.txt
|
|
```
|
|
|
|
4\) Install spconv
|
|
|
|
|
|
* Install the spconv dependencies.
|
|
```
|
|
sudo apt-get install libboost-all-dev
|
|
sudo apt-get install libsparsehash-dev
|
|
```
|
|
|
|
* Compile the spconv library.
|
|
```
|
|
cd SoftGroup/lib/spconv
|
|
python setup.py bdist_wheel
|
|
pip install dist/{WHEEL_FILE_NAME}.whl
|
|
```
|
|
|
|
|
|
5\) Compile the external C++ and CUDA ops.
|
|
```
|
|
cd SoftGroup/lib/softgroup_ops
|
|
python setup.py build_ext develop
|
|
```
|
|
|
|
Alternative installation guide can be found in [here](https://github.com/hustvl/HAIS).
|