-
在PyTorch中实现的空间稀疏卷积库spconv
资源介绍
SpConv:PyTorch空间稀疏卷积库
这是一个空间稀疏的卷积库,例如但速度更快且易于阅读。 该库提供稀疏卷积/转置,子流形卷积,逆卷积和稀疏maxpool。
2020-5-2,我们在此版本的spconv中添加了ConcatTable,JoinTable,AddTable和Identity函数,以构建ResNet和Unet。
码头工人:
docker pull scrin/dev-spconv ,包含python 3.8,cuda 10.1,鱼壳,最新的pytorch和tensorflow。
在Ubuntu 16.04 / 18.04上安装
如果您使用的是pytorch 1.4+,并且遇到“ nvcc致命:未知-Wall”,则需要转到割炬软件包目录并在Caffe2Targets.cmake的INTERFACE_COMPILE_OPTIONS中删除包含“ -Wall”的标志。 此项