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` 返回值的限制。
