在 MacBook Pro 上使用 PyTorch Cuda

Question

14

我正在使用 MacBook Pro (16 英寸, 2019, macOS 10.15.5 (19F96))

显卡

我尝试在我的 Mac 上使用带有 Cuda 的 Pytorch。

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

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

在 Mac 和 AMD 显卡上是否可能运行 Pytorch GPU?

- Kenny Smith

4个回答

10

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

要开始使用，请安装最新的PyTorch夜间版本: https://pytorch.org/get-started/locally/

不幸的是，在macOS上使用Pytorch时无法使用GPU加速。CUDA已经很长时间没有在macOS上可用，它只能在NVIDIA GPU上运行。AMD的等效库ROCm需要Linux。

如果您正在使用macOS 12.0或更高版本并愿意改用TensorFlow，则可以使用适用于Mac的优化版TensorFlow，该版本支持使用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

- Palle

2022年还正确吗？ - VimNing

正在进行将Metal支持引入Pytorch的工作，并且最新的夜间构建中似乎至少包含了一些对它的支持。然而，夜间构建很可能存在问题，不适合生产环境使用。 - Palle

6

CUDA是由nVidia开发的GPU计算框架，仅适用于nVidia GPU。CuDNN框架同理。

目前，使用PyTorch进行GPU加速需要nVidia GPU，即AMD GPU无法实现。操作系统不是问题，如macOS并不能解决这个问题，关键在于所使用的GPU。

但你可以购买外部nVidia GPU或使用一些集群。例如，Google Colab提供了PyTorch兼容性。

- Xxxo

2

预计在4个月内，即2022年3月左右将会实现。请参考Soumith在GitHub上对该问题的回答。https://github.com/pytorch/pytorch/issues/47702

- Somenzi

2

仅适用于M1 Macs... - yannis

似乎在我的2019年MacBook Pro上与RX5300M一起工作。 - Luca Di Liello

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Sergii Dymchenko · Accepted Answer

不可以。

CUDA仅支持NVidia GPU，不支持AMD GPU。

目前正在努力支持通过ROCm为PyTorch加速AMD GPU（但ROCm不适用于MacOS）。