在 MacBook Pro 上使用 pytorch Cuda

Question

我使用的是 MacBook Pro（16 英寸，2019，macOS 10.15.5 (19F96)）

GPU

• AMD Radeon Pro 5300M
• 英特尔超核芯显卡 630

我正在尝试在我的 Mac 上使用 Pytorch 和 Cuda。

我看到的所有指南都假设我有 Nvidia 显卡。

我发现了这个：https://github.com/pytorch/pytorch/issues/10657问题，但看起来我需要安装ROCm，并且根据他们的支持的操作系统，它只支持Linux。

是否可以使用 mac 和 AMD 显卡在 GPU 上运行 Pytorch？

Answer 1

没有。

CUDA 仅适用于受支持的 NVidia GPU，不适用于 AMD GPU。

我们正在努力支持使用 PyTorch 加速 AMD GPU（通过 ROCm，它不适用于 MacOS）。

Answer 2

PyTorch 现在支持使用 Metal 进行训练。

公告：https://pytorch.org/blog/introducing-accelerated-pytorch-training-on-mac/

首先，安装最新的每晚构建的 PyTorch：https://pytorch.org/get-started/locally/

---

2022 年 5 月前答复

不幸的是，在 macOS 上使用 Pytorch 时无法使用 GPU 加速。 CUDA 暂时无法在 macOS 上使用，并且只能在 NVIDIA GPU 上运行。 AMD 的等效库 ROCm 需要 Linux。

如果您使用的是 macOS 12.0 或更高版本，并且愿意使用 TensorFlow，则可以使用 TensorFlow 的 Mac 优化版本，它支持使用 Apple 自己的 GPU 加速库 Metal 进行 GPU 训练。

目前，您需要Python 3.8（<=3.7 and >=3.9 不起作用）才能运行它。要安装，请运行：

pip3 install tensorflow-macos
pip3 install tensorflow-metal

您可能需要先卸载现有的tensorflow发行版或在虚拟环境中工作。

那么你就可以

import tensorflow as tf

tf.test.is_gpu_available()  # should return True

Answer 3

CUDA 是一个 GPU 计算框架，由 nVidia 为 nVidia GPU 开发。此外，CuDNN 框架也是如此。

目前，您无法在具有 AMD GPU 的 PyTorch 中使用 GPU 加速，即没有 nVidia GPU。操作系统这不是问题，也就是说，你有 macOS 并不重要。这取决于你拥有什么 GPU。

您可以做的是，您可以购买外部 nVidia GPU 或使用某些集群。例如，Google Colab 提供 PyTorch 兼容性。

Answer 4

4 个月后，即 2022 年 3 月左右，就有可能。请参阅 Soumith 在 GitHub 上对此问题的回复。 https://github.com/pytorch/pytorch/issues/47702

Answer 5

2024年4月，Metal是使用MAc硬件加速的方式。 您可以使用

brew install torchvision

（如果您需要其他模块）或简单地使用

brew install pytorch

，但否则您将在 venv 中使用

来完成此操作

pip3 install torch torchvision

然后使用

import torch
if torch.backends.mps.is_available():
    mps_device = torch.device("mps")
    x = torch.ones(1, device=mps_device)
    print (x)
    # output expected:
    # tensor([1.], device='mps:0')

else:
    print ("MPS device not found.")

来源：https://developer.apple.com/metal/pytorch/