torch.mps.set_per_process_memory_fraction#
- torch.mps.set_per_process_memory_fraction(fraction)[原始碼]#
設定程序在 MPS 裝置上記憶體分配的記憶體分數。允許的值等於分數乘以推薦的最大裝置記憶體（從 Metal API device.recommendedMaxWorkingSetSize 獲取）。如果程序試圖分配的值超過允許值，則會在分配器中引發記憶體不足錯誤。
- 引數
fraction (float) – 範圍：0~2。允許的記憶體等於 total_memory * fraction。
注意
將 fraction 傳遞為 0 意味著無限制分配（如果記憶體不足，可能導致系統故障）。將 fraction 傳遞大於 1.0 的值允許超過 device.recommendedMaxWorkingSetSize 返回的值進行限制。
