MemPoolContext

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

MemPoolContext 保持当前活动的池并存储之前的池。删除时，它会使之前的池变为活动状态。

参数

• pool (torch.cuda.MemPool) – 要激活的 MemPool 对象，以便

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

static active_pool()

返回活动的 MemPool

返回类型

Optional[_MemPool]