要做毕设，要优化一个算法（不是机器学习相关的），这个算法本来只是利用Spark平台，导师建议把Spark和GPU结合起来，让Spark调用GPU资源进行计算，但是最近几天查了很多资料，方法似乎有很多，但是感觉可行的太少，目前情况是这样的：
原来这个算法是用python的pyspark实现的，所以我希望能找到相应的python库可以实现Spark调用GPU资源
找到一个numba+pyspark的，但是资料太少了，只有一个ppt，自己在运行ppt上代码时，出现了no module named numba.serialize的错误，怀疑是版本问题？
求问大家有没有Spark利用GPU资源的相应的包，最好是python的，因为可能没时间移植到别的语言了，已经实现的这个算法就是python的pyspark写的
