torch.mps.driver_allocated_memory¶ torch.mps.driver_allocated_memory()[原始碼][原始碼]¶ 傳回 Metal 驅動程式為該進程分配的 GPU 總記憶體，單位為位元組。 注意 傳回的大小包含 MPSAllocator 池中的快取分配，以及來自 MPS/MPSGraph 框架的分配。 傳回類型 int
