Python函数替代MATLAB的gpuArray（）

Question

MATLAB有一个很好的函数gpuArray();，用于利用GPU以闪电般的速度执行操作（特别是如果安装了一个好的显卡）。对于神经网络，这是必要的，因为输入矩阵很大并且执行了许多复杂的操作。

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

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

提前致谢！

Answer 1

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

http://documen.tician.de/pycuda/array.html