python - 如何在 Apple M1 芯片的 Mac 上使用 Tensorflow 检查 GPU 的可访问性？

Question

我在 Macbook（arm64，M1 芯片）上使用 Tensorflow 2.4.0，在我想检查 M1 芯片中的 GPU 是否可以被 Tensorflow 使用后，我得到了这个输出：

我的代码：

import tensorflow as tf
print(tf.test.gpu_device_name())
print(tf.config.list_physical_devices('GPU'))

输出：

[]

看起来我的 GPU 不可用。我应该如何在 M1 上使用我的 GPU 来加速训练？

Answer 1

自 2021 年 7 月起，Apple提供以下说明来安装 Tensorflow 2.5 和tensorflow-metal插件：

1. 创建一个新的conda环境；
2. 跑conda install -c apple tensorflow-deps
3. 安装张量流：python -m pip install tensorflow-macos；然后
4. 安装插件：python -m pip install tensorflow-metal.

这将使您能够访问 Tensorflow 中的 M1 GPU。

Answer 2

您可能需要为 macOS 11.0+ 安装这些 Tensorflow 插件。