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。

注意

将 0 传递给 fraction 表示不限制分配（可能导致内存不足时系统故障）。传递大于 1.0 的 fraction 值允许超出设备推荐的最大工作集大小的限制。