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