torch.mps.driver_allocated_memory

torch.mps.driver_allocated_memory()

返回 Metal 驅動程式為該程序分配的總 GPU 記憶體（以位元組為單位）。

注意

返回的大小包括 MPSAllocator 池中的快取分配以及來自 MPS/MPSGraph 框架的分配。

返回型別

int