如何在 Apple M2 上的 Pytorch 中运行 MPS？

Question

我有以下函数来检查我的 MacBook Pro Apple M2 Max 上的 Pytorch 中是否启用了 MPS。我得到回复：

MPS is not available
MPS is not built

然后我收到错误，因为如果 mps 不存在，MPS 无法检查 MacOS 的版本。

def check_mps():
    if torch.backends.mps.is_available():
        print("MPS is available")
    else:
        print("MPS is not available")

    if torch.backends.mps.is_built():
        print("MPS is built")
    else:
        print("MPS is not built")

    if torch.backends.mps.is_macos_or_newer(13, 0):
        print("MPS is macOS 13 or newer")
    else:
        print("MPS is not macOS 13 or newer")

我已经验证了我的操作系统是最新的，并且我没有看到让 Pytorch 在 Metal 上运行的许多其他步骤。我错过了什么？

我尝试更新我的操作系统，这是迄今为止我找到的唯一建议。它没有改变任何东西。

Answer 1

经过更多研究，我发现 mps 仅内置于 pytorch nightly 版本中，可通过以下方式安装：

conda install pytorch-nightly::pytorch -c pytorch-nightly

https://pytorch.org/get-started/locally/