Numpy 可以针对不同的 BLAS 实现（MKL、ACML、ATLAS、GotoBlas 等）“链接/编译”。这并不总是很容易配置，但它是可能的。
是否也可以针对 NVIDIA 的 CUBLAS 实现“链接/编译”numpy？
我在网络上找不到任何资源，在我花太多时间尝试之前，我想确保它完全可行。
问问题
12595 次
2 回答
19
一句话：不，你不能那样做。
有一个相当不错的 scikit，它提供了从 scipy 访问 CUBLAS 的功能scikits.cuda，它建立在PyCUDA 之上。PyCUDA 提供了一个numpy.ndarray类似的类，它允许使用 CUDA 无缝地操作 GPU 内存中的 numpy 数组。所以你可以将 CUBLAS 和 CUDA 与 numpy 一起使用，但你不能只链接到 CUBLAS 并期望它能够工作。
还有一个商业库，它提供类似 numpy 和 cublas 的功能，并且具有 Python 接口或绑定，但我将把它留给他们的一个工作人员来填补这一点。
于 2012-07-20T09:51:34.550 回答
4
这是另一种可能性：
http://www.cs.toronto.edu/~tijmen/gnumpy.html
这基本上是一个 gnumpy + cudamat 环境，可用于利用 GPU。同样的代码也可以在没有 gpu 的情况下使用 npmat 运行。请参阅上面的链接以下载所有这些文件。
于 2013-09-27T16:52:25.013 回答