了解 Mac OS MPS 警告

了解 Mac OS MPS 警告

语境

  • 硬件:Mac M1 Mini
  • 软件
% sw_vers
ProductName:        macOS
ProductVersion:     13.5
BuildVersion:       22G74

手头的任务:运行一段 Python 代码(机器学习模型)。

我在 CPU 上运行它并且确实执行良好,所以代码不是问题。

在我的 M1 的 GPU 上运行它时,程序崩溃并出现以下错误:

RuntimeError: MPS backend out of memory (MPS allocated: 5.34 GB, other allocations: 2.95 GB, max allowed: 9.07 GB). Tried to allocate 1.32 GB on private pool. Use PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0 to disable upper limit for memory allocations (may cause system failure).

我需要帮助来了解有关该错误的一些事情,以及修复该错误的任何建议:

  1. 我使用 powermetrics 检查了 GPU 活动,据我所知,它目前没有运行:

**** GPU usage ****

GPU HW active frequency: 0 MHz
GPU HW active residency:   0.00% (396 MHz:   0% 528 MHz:   0% 720 MHz:   0% 924 MHz:   0% 1128 MHz:   0% 1278 MHz:   0%)
GPU SW requested state: (P1 :   0% P2 :   0% P3 :   0% P4 :   0% P5 :   0% P6 :   0%)
GPU SW state: (SW_P1 :   0% SW_P2 :   0% SW_P3 :   0% SW_P4 :   0% SW_P5 :   0% SW_P6 :   0%)
GPU idle residency: 100.00%
GPU Power: 0 mW
  1. 关于如何修复或检查这个问题有什么想法吗?

相关内容