cupy.cuda.malloc_managed

cupy.cuda.malloc_managed(size_t size) → MemoryPointer

分配托管内存（统一内存）。

此方法可用作 CuPy 内存分配器。将托管内存用作默认分配器的最简单方法如下所示：

set_allocator(malloc_managed)

在 CuPy 中使用托管内存的优点是，对于设备属性 cudaDevAttrConcurrentManagedAccess 值不为零的 GPU，可以实现设备内存超额订阅。建议使用 CUDA >= 8.0 和 Pascal 或更高版本的 GPU。

更多信息请参阅：https://docs.nvda.net.cn/cuda/cuda-runtime-api/group__CUDART__MEMORY.html#axzz4qygc1Ry1 # NOQA

参数:

size (int) – 要分配的内存大小（字节）。

返回:

指向已分配缓冲区的指针。

返回类型:

MemoryPointer