模块“火炬”没有属性“has_mps”

Question

我刚刚观看了一段 youtube 视频，该视频教授如何在夜间为 MacBook 安装 PyTorch 以通过 m1 芯片加速。但是，我遇到了一个非常奇怪的问题。我可以在 jupyter 笔记本中看到 torch.has_mps = True。但是在 vscode 的 jupyter notebook 中，它显示模块 'torch' 没有属性 'has_mps'。

谁能告诉我为什么？真的很混乱。

Answer 1

只需确保您安装了 PyTorch 的夜间版本。 PyTorch 中的 Apple Silicon 支持目前仅在夜间构建中可用。

例如，如果你使用 conda，试试这个：

conda install pytorch torchvision -c pytorch-nightly

或者用点子

pip3 install --pre torch torchvision --extra-index-url https://download.pytorch.org/whl/nightly/cpu

在这里查看更多：

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

验证您是否使用正确版本的脚本就像您编写的那样，您只需在安装了上述内容的环境中打开 python REPL：

import torch
torch.has_mps

你应该得到

True

附：虽然指南建议安装torchaudio，但它不会起作用，至少在conda 环境下是这样。