torch.mps.driver_allocated_memory¶ torch.mps.driver_allocated_memory()[源代码]¶ 返回Metal驱动程序为进程分配的总GPU内存（以字节为单位）。 注意 返回的大小包括MPSAllocator池中的缓存分配以及来自MPS/MPSGraph框架的分配。 Return type int
