在 MacBook Pro 上使用 pytorch Cuda

Question

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

图形处理器

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

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

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

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

Answer 1

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

首先，安装最新的 PyTorch 夜间版本：https://pytorch.org/get-started/locally/

不幸的是，在 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 和 >=3.9 不起作用）才能运行它。要安装，请运行：

pip3 install tensorflow-macos
pip3 install tensorflow-metal

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

然后你就可以

import tensorflow as tf

tf.test.is_gpu_available()  # should return True

Answer 2

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

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

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

Answer 3

不。

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

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