python - Python 函数替代 MATLAB 的 gpuArray()

Question

MATLAB有一个很好的功能gpuArray();，可以利用 GPU 以闪电般的速度执行操作（尤其是在安装了好显卡的情况下）。对于神经网络，这是必要的，因为输入矩阵很大，并且执行了许多复杂的操作。

在matlab中，这只是通过这段代码完成的， G = gpuArray(ones(100, 'uint32'));

Python中有类似的东西吗？也就是说，一个像 MATLAB 的 GPU 库一样易于使用的开源库。

提前致谢！

score 1 · Accepted Answer

1

我想你会在 PyCUDA 中找到你想要的东西：

于 2013-07-07T20:19:35.033 回答

1 回答 1