CUDA内核上的SciPy编程

Question

有人设法在CUDA内核上应用SciPy函数吗？据我所知，SciPy没有明确支持GPU计算。

我想通过将某些SciPy.optimize函数应用于CUDA内核来加快代码的速度。如果有人知道可用于CUDA GPU的任何其他非线性优化库，请在此处共享。

Answer 1

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

Answer 2

遇到了类似的问题，我现在正在使用tensorflow_probability's optimize capabilities。最大的问题是tensorflow ops通常无法直接从numpy函数实现，因此学习曲线相当陡峭。但是，一旦编写了代码，它就会非常快。