这个问题是因为 PyTorch GPU 运算时使用了 mps (multipeer connectivity)功能导致的内存泄漏。解决这个问题的方法是禁用 mps。可以通过在代码中添加以下语句来禁用 mps:
import torch torch.cuda.set_enabled_lms(False)
添加这段代码后,PyTorch 运算就会在普通的 GPU 上进行,而不会使用 mps 的多个 GPU 连接功能。这样就可以避免内存泄漏问题。
上一篇:AppleM1Pro芯片中GoogleCastSDK不工作
下一篇:AppleM1PyTorchmps内存泄漏?