torch.cuda.change_current_allocator

torch.cuda.change_current_allocator(allocator)[source][source]

更改当前使用的内存分配器为提供的分配器。

如果当前分配器已被使用/初始化，此函数将报错。

参数:

allocator (torch.cuda.memory._CUDAAllocator) – 要设置为活动状态的分配器。

注意

详细说明创建和使用自定义分配器的内存管理。