PyTorch 中的 Apple M1/M2 GPU 支持：向前迈出了一步，但比传统的 Nvidia GPU 方法慢_apple gpu nvidia

我在 2021 年初购买了 Macbook Air M1 芯片。它速度快、重量轻，但无法利用 GPU 进行深度学习。好在我的数据集比较小，8核CPU就够用了。最近，我一直在做另一个项目，训练速度比预期低很多，所以我再次google了一下在M1/M2芯片上使用GPU。事实证明，PyTorch 去年发布了一个名为 Nightly 的新版本，允许在 Mac 上使用 GPU。我尝试了一下，发现使用 Nvidia GPU 仍然更好。

安装

性能对比

1. 安装

您需要将系统更新到 macOS 12.3+ 并安装这个最新的软件包。

我使用 Conda 来创建环境。

conda create -n myenv python=3.8
conda activate myenv
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
1
2
3

之后，您可以使用此代码检查 MPS 是否已激活。

import torch
import math

print(torch.backends.mps.is_available()) #the MacOS is higher than 12.3+
print(torch.backends.mps.is_built
1
2
3
4