MemPoolContext

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

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

参数

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

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

static active_pool()

返回活动 MemPool

返回类型

Optional[_MemPool]