cupy.get_default_memory_pool¶
-
cupy.
get_default_memory_pool
()[source]¶ Returns CuPy default memory pool for GPU memory.
Returns: The memory pool object. Return type: cupy.cuda.MemoryPool Note
If you want to disable memory pool, please use the following code.
>>> cupy.cuda.set_allocator(None)