在MacBook Pro上使用pytorch

Question

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

GPU

• AMD Radeon Pro 5300 m
• Intel UHD Graphics 630

我正试着在我的mac上使用Pytorch和Cuda。

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

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

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

Answer 1

不是的。

CUDA只与受支持的NVidia GPU一起工作，而不是与AMD一起工作。

有一个持续的努力，以支持加速与PyTorch (通过ROCm，它不工作在MacOS)的AMD GPU。

Answer 2

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

目前，您不能使用GPU加速与PyTorch与AMD ，即没有nVidia GPU。O.S.不是问题所在，也就是说，拥有macOS并不重要。这是你有什么GPU的问题。

您可以做的是，您可以购买一个外部nVidia图形处理器或使用一些集群。例如，Google提供了PyTorch兼容性。

Answer 3

PyTorch现在支持使用金属进行培训。

公告：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上。AMD的等价库ROCm需要Linux。

如果您正在使用macOS 12.0或更高版本，并且愿意使用TensorFlow，则可以使用TensorFlow的优化构建，该版本支持使用苹果自己的GPU加速库TensorFlow进行GPU培训。

目前，您需要Python3.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