SpConv: 空间稀疏卷积库
spconv Spatial Sparse Convolution Library 项目地址: https://gitcode.com/gh_mirrors/sp/spconv
项目基础介绍和主要编程语言
SpConv(Spatial Sparse Convolution Library)是一个高度优化的稀疏卷积库,主要用于处理包含大量零元素的稀疏数据,如3D点云数据。该项目由Traveller59开发并维护,主要使用C++和Python进行开发。SpConv的设计旨在提供高效的稀疏卷积实现,支持Tensor Core,适用于需要高性能计算的场景。
项目核心功能
SpConv的核心功能包括:
- 稀疏卷积操作:提供高度优化的稀疏卷积实现,支持多种CUDA版本,包括CUDA 10.2、11.3、11.4、11.7、11.8和12.0。
- Tensor Core支持:利用NVIDIA的Tensor Core技术,显著提升计算性能,特别是在支持Tensor Core的GPU上。
- 跨平台支持:支持Linux和Windows操作系统,适用于多种开发环境。
- 易于安装:通过PyPI提供预编译的二进制包,用户可以通过简单的
pip install
命令进行安装。
项目最近更新的功能
SpConv最近的更新包括:
- SpConv 2.3:新增了对int8量化的支持,用户可以通过文档和示例了解更多细节。
- SpConv 2.2:增加了对Ampere架构的支持,包括纯C++代码生成、NVRTC支持等。在Ampere GPU上,SpConv 2.2相比SpConv 2.1在fp16卷积核上提升了5-30%的性能,在int8卷积核上提升了1.2x-2.7x的性能。
- SpConv 2.1:引入了对Microsoft Windows的支持,用户无需手动构建,可以通过pip直接安装。此外,SpConv 2.1在fp32和fp16训练/推理速度上均有显著提升,特别是在支持Tensor Core的层上。
通过这些更新,SpConv不仅在性能上有了显著提升,还增强了跨平台和易用性,使其成为处理稀疏数据的首选工具。
spconv Spatial Sparse Convolution Library 项目地址: https://gitcode.com/gh_mirrors/sp/spconv