我尝试在我的 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 个月内发布公开 Alpha 版。
我们会尽快开放这个后端的开发。
那个帖子： https://github.com/pytorch/pytorch/issues/47702#issuecomment-965625139
TL;DR：公开测试版至少需要 4 个月。
【讨论】：