cupy.get_default_memory_pool#

cupy.get_default_memory_pool()[源代码][源代码]#

返回用于GPU内存的Cupy默认内存池。

返回:

内存池对象。

返回类型:

cupy.cuda.MemoryPool

备注

如果你想禁用内存池,请使用以下代码。

>>> cupy.cuda.set_allocator(None)