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.nvidia.com/cuda/cuda-runtime-api/group__CUDART__MEMORY.html#axzz4qygc1Ry1 # NOQA

参数:

size (int) – 内存分配的大小，单位为字节。

返回:

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

返回类型:

MemoryPointer