基于pytorch实现的softnms怎么用gpu加速

运行softnms的时候非常慢，结果发现GPU根本没用上，一查才发现要编译什么的，求助！