torch.mps.driver_allocated_memory

torch.mps.driver_allocated_memory()[源代码]

返回Metal驱动程序为此进程分配的总GPU内存（以字节为单位）。

注意

返回的大小包含了 MPSAllocator 池中的缓存分配，以及来自 MPS 和 MPSGraph 框架的分配。

返回类型

int