MemPoolContext

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

MemPoolContext 儲存當前的活躍記憶體池，並暫存之前的記憶體池。刪除時，它會將之前的記憶體池設為活躍。

引數

• pool (torch.cuda.MemPool) – 一個 MemPool 物件，將被設為活躍，以便

• pool. (分配路由到此處) –

static active_pool()

返回活躍的 MemPool

返回型別

Optional[_MemPool]