如何在 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

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