torch.mps.set_per_process_memory_fraction¶
- torch.mps.set_per_process_memory_fraction(fraction)[源代码]¶
设置内存分数以限制进程在MPS设备上的内存分配。 允许的值等于分数乘以推荐的最大设备内存 （从Metal API device.recommendedMaxWorkingSetSize获取）。 如果在进程中尝试分配超过允许值的内存，它将在分配器中引发内存不足错误。
- Parameters
fraction (float) – 范围: 0~2。允许的内存等于总内存 * fraction。
注意
将0传递给fraction表示无限分配（如果内存不足，可能会导致系统故障）。 传递大于1.0的fraction允许超出从device.recommendedMaxWorkingSetSize返回的值的限制。
