torch.cuda.set_per_process_memory_fraction¶
- torch.cuda.set_per_process_memory_fraction(fraction, device=None)[source][source]¶
設定程序的記憶體比例。
此比例用於限制快取分配器在 CUDA 裝置上分配的記憶體。允許的值等於總可見記憶體乘以該比例。如果在程序中嘗試分配超過允許值，分配器將引發記憶體不足錯誤。
- 引數
fraction (float) – 範圍：0~1。允許的記憶體等於 total_memory * fraction。
device (torch.device 或 int, 可選) – 所選裝置。如果為
None，則使用預設 CUDA 裝置。
注意
通常，總可用空閒記憶體小於總容量。
