问题描述
我试图在我的Macbook PRO上使用PyTorch训练一个模型。它使用的是新一代苹果M1 CPU。但是，PyTorch无法识别我的GPU。
GPU available: False, used: False
TPU available: False, using: 0 TPU cores
IPU available: False, using: 0 IPUs
有人知道解决方案吗？
我已将所有库更新到最新版本。
推荐答案
似乎正在为M1 GPU提供PyTorch支持，但尚未完成。
来自GitHub：
上的@soumith所以，这是最新消息。 我们计划支持M1 GPU。@albanD、@ezYang和一些核心开发人员一直在调查这件事。我现在不能确认/否认任何其他人的参与。
那么，到目前为止，我们得到的是一个原型，它还可以。我们采用了错误的方法(更多的是图形匹配)，用户体验不是很好--有些操作非常快，有些非常慢，总体体验并不顺畅。人们必须猜测他们的哪些工作流会更快。所以，我们正在使用一种新的方法完全重写它，我认为它更接近于您的Ole PyTorch，但这需要一些时间。我认为我们不会在接下来的4个月内达到公开的阿尔法。
我们将尽快开放此后端的开发。
该帖子： https://github.com/pytorch/pytorch/issues/47702#issuecomment-965625139TL；DR：公开测试版至少还有4个月。