torch.mps.set_per_process_memory_fraction#
- torch.mps.set_per_process_memory_fraction(fraction)[source]#
设置内存比例以限制进程在MPS设备上的内存分配。允许的值等于比例乘以从Metal API device.recommendedMaxWorkingSetSize获得的最大设备内存。如果在进程中尝试分配超出允许值的内容，则分配器会引发内存不足错误。
- 参数
fraction (float) – 范围：0~2。允许的内存等于total_memory * fraction。
注意
将fraction设置为0表示无限分配（如果内存不足，可能导致系统故障）。将fraction设置为大于1.0的值允许超出device.recommendedMaxWorkingSetSize返回值范围的限制。