如何在Macbook PRO(M1)GPU上运行Pytorch？ [英] How to run Pytorch on Macbook pro (M1) GPU?

问题描述

我试图在我的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-965625139

TL；DR：公开测试版至少还有4个月。

这篇关于如何在Macbook PRO(M1)GPU上运行Pytorch？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！