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。允許的記憶體等於 total_memory * fraction。
注意
將 0 傳遞給 fraction 表示無限分配（如果記憶體不足，可能會導致系統故障）。傳遞大於 1.0 的 fraction 允許超出 device.recommendedMaxWorkingSetSize 傳回的值的限制。
