MemPoolContext

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

MemPoolContext 保存当前的活跃内存池，并暂存之前的内存池。删除时，它会将之前的内存池设为活跃。

参数

• pool (torch.cuda.MemPool) – 一个 MemPool 对象，将被设为活跃，以便

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

static active_pool()

返回活跃的 MemPool

返回类型

Optional[_MemPool]