快捷方式

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源