torch.mps.set_per_process_memory_fraction

torch.mps.set_per_process_memory_fraction(fraction)

设置 MPS 设备上进程内存分配的内存分数。允许的值等于分数乘以推荐的最大设备内存（从 Metal API 的 device.recommendedMaxWorkingSetSize 获取）。如果尝试在一个进程中分配超过允许的值，将在分配器中引发内存不足错误。

参数:

分数（浮点数）- 范围：0~2。允许的内存等于总内存 * 分数。

注意

将 0 传递给分数表示无限制分配（如果内存不足可能导致系统故障）。传递大于 1.0 的分数允许超过 device.recommendedMaxWorkingSetSize 返回的值。