一直以来，Pytorch 在 Mac 上仅支持使用 CPU 进行训练。就在刚刚，Pytorch 官方宣布，其最新版 v1.12 可以支持 GPU 加速了。只要是搭载了 M1 系列芯片的 Mac 都行。

　　Metal 是一个类似 OpenGL 的框架，只不过 OpenGL 适用于各平台的移动端 GPU 渲染和计算，Metal专用于 iOS / MacOS 平台，不过也兼顾了性能和易用性。

　　MPS 就是一套基于 Metal 框架的库，直接调用即可使用 GPU 的高性能进行图形处理、构建卷积神经网络等工作小火箭mac端怎么连接。

　　从下图中我们可以发现，相比使用 CPU 加速，使用 GPU 可将模型训练速度提高约 7 倍，评估（evaluation）速度则最高能提约 20 倍。

　　有人表示，虽说目前 M1 的原始计算性能比不上英伟达的产品，但功耗方面还不错。未来苹果很有可能慢慢追上性能。总的来说，Mac Studio 现在看起来实在太香了。

　　他进一步解释道：“毕竟它是你花 4800 美元就能买到的最便宜、包含 128GB GPU 内存的机器。现在有了基于 GPU 加速的 PyTorch 支持，完全可以用来训练大模型、配置大的 batch size。对于我所做的那种 DL 工作，数据加载比实际的原始计算能力更容易成为瓶颈。”

　　只需保证你的 macOS 操作系统在 12.3 版本及以上，且安装了 arm64 原生 Python，然后去官网下载最新的 Pytorch 预览版就可以了。

　　广告声明：文内含有的对外跳转链接（包括不限于超链接、二维码、口令等形式），用于传递更多信息，节省甄选时间，结果仅供参考，IT之家所有文章均包含本声明。