MemPool

class torch.cuda.MemPool(*args, **kwargs)

MemPool 代表缓存分配器中的一组内存。目前，它只是在 CUDACachingAllocator 中维护的池对象的 ID。

参数

allocator (torch._C._cuda_CUDAAllocator, 可选) – 一个 torch._C._cuda_CUDAAllocator 对象，可用于定义如何在池中分配内存。如果 allocator 为 None（默认值），则内存分配遵循 CUDACachingAllocator 的默认/当前配置。

property allocator: Optional[_cuda_CUDAAllocator]

返回此 MemPool 将分配路由到的分配器

property id: Tuple[int, int]

将此池的 ID 作为两个整数的元组返回。