cupy.get_default_memory_pool#

cupy.get_default_memory_pool()[源码]#

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

返回值:

内存池对象。

返回类型:

cupy.cuda.MemoryPool

注意

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

>>> cupy.cuda.set_allocator(None)