CUDA核上的SciPy编程

Question

是否有人成功地在CUDA内核上应用了SciPy函数？据我所知，从文献中看，没有明确支持SciPy的图形处理器计算。

我想通过在CUDA内核上应用一些SciPy.optimize函数来加速代码。如果有人知道可以在CUDA GPU上应用的任何其他非线性优化库，将非常感谢在这里分享它。

Answer 1

Numba可能就是你要找的，它允许你在GPU上运行python代码。另一个好的资源可能是this tutorial

Answer 2

遇到过类似的问题，我现在使用的是tensorflow_probability's optimize capabilities。最大的问题是tensorflow操作通常不能直接从numpy函数实现，因此有一个非常陡峭的学习曲线。但是一旦你的代码写好了，它就会变得非常快。