numpy - Numpy、BLAS 和 CUBLAS

Question

Numpy 可以针对不同的 BLAS 实现（MKL、ACML、ATLAS、GotoBlas 等）“链接/编译”。这并不总是很容易配置，但它是可能的。

是否也可以针对 NVIDIA 的 CUBLAS 实现“链接/编译”numpy？
我在网络上找不到任何资源，在我花太多时间尝试之前，我想确保它完全可行。

Answer 1

一句话：不，你不能那样做。

有一个相当不错的 scikit，它提供了从 scipy 访问 CUBLAS 的功能scikits.cuda，它建立在PyCUDA 之上。PyCUDA 提供了一个numpy.ndarray类似的类，它允许使用 CUDA 无缝地操作 GPU 内存中的 numpy 数组。所以你可以将 CUBLAS 和 CUDA 与 numpy 一起使用，但你不能只链接到 CUBLAS 并期望它能够工作。

还有一个商业库，它提供类似 numpy 和 cublas 的功能，并且具有 Python 接口或绑定，但我将把它留给他们的一个工作人员来填补这一点。

Answer 2

这是另一种可能性：

http://www.cs.toronto.edu/~tijmen/gnumpy.html

这基本上是一个 gnumpy + cudamat 环境，可用于利用 GPU。同样的代码也可以在没有 gpu 的情况下使用 npmat 运行。请参阅上面的链接以下载所有这些文件。