Cupy 将 numpy 数组复制到现有设备数组

Question

我想在现有的预分配 GPU 数组上复制 numpy 数组。

我看到cupy提供了函数

copy

和

copyto

，但是前者不允许指定目标数组，而后者只能在设备数组之间工作。

有什么方法可以利用预先分配的内存将 numpy 数组复制到设备吗？

Answer 1

paleonix 的建议确实有效，即如下所示：

cupy.cuda.runtime.memcpy(cupy_array.data.ptr, numpy_array.ctypes.data, numpy_array.nbytes, cupy.cuda.runtime.memcpyHostToDevice)