torch.mps.set_per_process_memory_fraction

torch.mps.set_per_process_memory_fraction(fraction)

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

参数

fraction (float) – 范围：0~2。允许的内存等于 total_memory * fraction。

注意

将 fraction 传递为 0 意味着无限制分配（如果内存不足，可能导致系统故障）。将 fraction 传递大于 1.0 的值允许超过 device.recommendedMaxWorkingSetSize 返回的值进行限制。