内存池上下文 ¶

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

MemPoolContext 保存当前活动的池并存储上一个池。删除时使上一个池变为活动状态。

参数:

• pool (torch.cuda.MemPool) – 一个要激活的 MemPool 对象

• pool.（分配路由到此处） –

static active_pool()

返回活动 MemPool

返回类型:

可选[内存池]