mirror of
https://github.com/botastic/SoftGroup.git
synced 2025-10-16 11:45:42 +00:00
41 lines
727 B
Markdown
41 lines
727 B
Markdown
## Installation
|
|
|
|
1\) Environment requirements
|
|
|
|
* Python 3.x
|
|
* Pytorch 1.11
|
|
* CUDA 9.2 or higher
|
|
|
|
The following installation guild suppose ``python=3.7`` ``pytorch=1.11`` and ``cuda=10.2``. You may change them according to your system.
|
|
|
|
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
|
|
```
|
|
|
|
|
|
3\) Install the dependencies.
|
|
```
|
|
conda install pytorch cudatoolkit=10.2 -c pytorch
|
|
pip install spconv-cu102
|
|
pip install -r requirements.txt
|
|
```
|
|
|
|
4\) Install build requirement.
|
|
|
|
```
|
|
sudo apt-get install libsparsehash-dev
|
|
```
|
|
|
|
5\) Setup
|
|
```
|
|
python setup.py build_ext develop
|
|
```
|