Need help with spconv?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

traveller59
768 Stars 220 Forks Apache License 2.0 174 Commits 47 Opened issues

Description

Spatial Sparse Convolution Library

Services available

!
?

Need anything else?

Contributors list

# 25,840
Python
Jupyter...
CSS
object-...
50 commits
# 21,075
C++
pytorch
3d-obje...
MATLAB
5 commits
# 383,634
Python
Shell
4 commits
# 514,748
Python
Shell
1 commit
# 516,398
Python
Shell
1 commit

SpConv: Spatially Sparse Convolution Library

Build Status

| | PyPI | Install |Downloads | | -------------- |:---------------------:| ---------------------:| ---------------------:| | CPU (Linux Only) | PyPI Version |

pip install spconv
| pypi monthly download | | CUDA 10.2 | PyPI Version |
pip install spconv-cu102
| pypi monthly download | | CUDA 11.1 | PyPI Version |
pip install spconv-cu111
| pypi monthly download| | CUDA 11.3 (Linux Only) | PyPI Version |
pip install spconv-cu113
| pypi monthly download| | CUDA 11.4 | PyPI Version |
pip install spconv-cu114
| pypi monthly download|

spconv
is a project that provide heavily-optimized sparse convolution implementation with tensor core support. check benchmark to see how fast spconv 2.x runs.

Spconv 1.x code. We won't provide any support for spconv 1.x since it's deprecated. use spconv 2.x if possible. <!--remove this message in spconv 2.2-->

Check spconv 2.x algorithm introduction to understand sparse convolution algorithm in spconv 2.x!

WARNING spconv < 2.1.18 users need to upgrade your version to 2.1.18, it fix a bug in conv weight init which cause std of inited weight too large, and a bug in PointToVoxel.

Breaking changes in Spconv 2.x

Spconv 1.x users NEED READ THIS before using spconv 2.x.

Spconv 2.1 vs Spconv 1.x

  • spconv now can be installed by pip. see install section in readme for more details. Users don't need to build manually anymore!
  • Microsoft Windows support (only windows 10 has been tested).
  • fp32 (not tf32) training/inference speed is increased (+50~80%)
  • fp16 training/inference speed is greatly increased when your layer support tensor core (channel size must be multiple of 8).
  • int8 op is ready, but we still need some time to figure out how to run int8 in pytorch.
  • doesn't depend on pytorch binary, but you may need at least pytorch >= 1.5.0 to run spconv 2.x.
  • since spconv 2.x doesn't depend on pytorch binary (never in future), it's impossible to support torch.jit/libtorch inference.

Spconv 2.x Development and Roadmap

Spconv 2.2 development has started. See this issue for more details.

See dev plan. A complete guide of spconv development will be released soon.

Usage

Firstly you need to use

import spconv.pytorch as spconv
in spconv 2.x.

Then see this.

Don't forget to check performance guide.

Install

You need to install python >= 3.6 (>=3.7 for windows) first to use spconv 2.x.

You need to install CUDA toolkit first before using prebuilt binaries or build from source.

You need at least CUDA 10.2 to build and run spconv 2.x. We won't offer any support for CUDA < 10.2.

Prebuilt

We offer python 3.6-3.10 and cuda 10.2/11.1/11.3/11.4 prebuilt binaries for linux (manylinux).

We offer python 3.7-3.10 and cuda 10.2/11.1/11.4 prebuilt binaries for windows 10/11.

We will provide prebuilts for CUDA versions supported by latest pytorch release. For example, pytorch 1.10 provide cuda 10.2 and 11.3 prebuilts, so we provide them too.

For Linux users, you need to install pip >= 20.3 first to install prebuilt.

CUDA 11.1 will be removed in spconv 2.2 because pytorch 1.10 don't provide prebuilts for it.

pip install spconv
for CPU only (Linux Only). you should only use this for debug usage, the performance isn't optimized due to manylinux limit (no omp support).

pip install spconv-cu102
for CUDA 10.2

pip install spconv-cu111
for CUDA 11.1

pip install spconv-cu113
for CUDA 11.3 (Linux Only)

pip install spconv-cu114
for CUDA 11.4

NOTE It's safe to have different minor cuda version between system and conda (pytorch) in CUDA >= 11.0 because of CUDA Minor Version Compatibility. For example, you can use spconv-cu114 with anaconda version of pytorch cuda 11.1 in a OS with CUDA 11.2 installed.

For CUDA 10, we don't know whether

spconv-cu102
works with CUDA 10.0 and 10.1. Users can have a try.

NOTE In Linux, you can install spconv-cuxxx without install CUDA to system! only suitable NVIDIA driver is required. for CUDA 11, we need driver >= 450.82.

Prebuilt GPU Support Matrix

See this page to check supported GPU names by arch.

| CUDA version | GPU Arch List | | -------------- |:---------------------:| | 10.2 | 50,52,60,61,70,75 | | 11.x | 52,60,61,70,75,80,86 | | 12.x | 60,61,70,75,80,86,90 |

Build from source for development (JIT, recommend)

The c++ code will be built automatically when you change c++ code in project.

For NVIDIA Embedded Platforms, you need to specify cuda arch before build:

export CUMM_CUDA_ARCH_LIST="7.2"
for xavier,
export CUMM_CUDA_ARCH_LIST="6.2"
for TX2,
export CUMM_CUDA_ARCH_LIST="8.7"
for orin.

You need to remove

cumm
in
requires
section in pyproject.toml after install editable
cumm
and before install spconv due to pyproject limit (can't find editable installed
cumm
).

You need to ensure

pip list | grep spconv
and
pip list | grep cumm
show nothing before install editable spconv/cumm.

Linux

  1. uninstall spconv and cumm installed by pip
  2. install build-essential, install CUDA
  3. git clone https://github.com/FindDefinition/cumm
    ,
    cd ./cumm
    ,
    pip install -e .
  4. git clone https://github.com/traveller59/spconv
    ,
    cd ./spconv
    ,
    pip install -e .
  5. in python,
    import spconv
    and wait for build finish.

Windows

  1. uninstall spconv and cumm installed by pip
  2. install visual studio 2019 or newer. make sure C++ development component is installed. install CUDA
  3. set powershell script execution policy
  4. start a new powershell, run
    tools/msvc_setup.ps1
  5. git clone https://github.com/FindDefinition/cumm
    ,
    cd ./cumm
    ,
    pip install -e .
  6. git clone https://github.com/traveller59/spconv
    ,
    cd ./spconv
    ,
    pip install -e .
  7. in python,
    import spconv
    and wait for build finish.

Build wheel from source (not recommend, this is done in CI.)

You need to rebuild

cumm
first if you are build along a CUDA version that not provided in prebuilts.

Linux

  1. install build-essential, install CUDA
  2. run
    export SPCONV_DISABLE_JIT="1"
  3. run
    pip install pccm cumm wheel
  4. run
    python setup.py bdist_wheel
    +
    pip install dists/xxx.whl

Windows

  1. install visual studio 2019 or newer. make sure C++ development component is installed. install CUDA
  2. set powershell script execution policy
  3. start a new powershell, run
    tools/msvc_setup.ps1
  4. run
    $Env:SPCONV_DISABLE_JIT = "1"
  5. run
    pip install pccm cumm wheel
  6. run
    python setup.py bdist_wheel
    +
    pip install dists/xxx.whl

Know issues

  • Spconv 2.x F16 runs slow in A100.

Note

The work is done when the author is an employee at Tusimple.

LICENSE

Apache 2.0

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.