torch.mps.recommended_max_memory

torch.mps.recommended_max_memory()

返回推荐的最大工作集大小，以字节为单位，用于 GPU 内存。

注意

Metal 推荐的最高工作集大小。由 device.recommendedMaxWorkingSetSize 返回。

返回类型:

int