MemPoolContext

classtorch.cuda.MemPoolContext(*args, **kwargs)[源代码]

MemPoolContext 保存当前活跃的内存池，并保留之前的内存池。在被删除时，它会使之前的内存池变为活跃状态。

参数

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

• 池。 （分配路由至此） –

静态active_pool()[源代码]

返回活动的内存池

返回类型

Optional[_MemPool]