cupy.cuda.set_allocator

cupy.cuda.set_allocator(allocator=None)

设置当前 GPU 内存分配器。

参数：

分配器 (function) – CuPy 内存分配器。它必须具有与 cupy.cuda.alloc() 函数相同的接口，该函数接受缓冲区大小作为参数，并返回该大小的设备缓冲区。指定为 None 时，将使用原始内存分配器（即禁用内存池）。