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]

以兩個 int 組成的元組形式返回此池的 ID。

snapshot()

返回所有裝置上 CUDA 記憶體分配器池狀態的快照。

解釋此函式的輸出需要熟悉記憶體分配器內部細節。

注意

有關 GPU 記憶體管理的更多詳細資訊，請參閱記憶體管理。

use_count()

返回此記憶體池的引用計數。

返回型別

int