torch.mps.set_per_process_memory_fraction¶
- torch.mps.set_per_process_memory_fraction(fraction)[源代码][源代码]¶
设置用于限制进程在 MPS 设备上内存分配的内存比例。允许的值等于该比例乘以 MPS 建议的最大设备内存（从 Metal API `device.recommendedMaxWorkingSetSize` 获取）。如果进程尝试分配超过允许值的内存，则分配器将抛出内存不足错误。
- 参数
fraction (float) – 范围：0~2。允许内存等于 total_memory * fraction。
注意
将 fraction 设置为 0 意味着无限制分配（如果内存不足可能会导致系统故障）。将 fraction 设置为大于 1.0 的值允许超过 `device.recommendedMaxWorkingSetSetSize` 返回值的限制。
