无法识别M1 MacBook上平台GPU 0的NUMA节点

Question

我无法识别下面的警告。我用的是M1 MacBook Air

码

from keras import models
from keras.layers import Dense

model=models.Sequential()
model.add(Dense(16,activation='relu',input_shape=(10000,)))
model.add(Dense(16,activation='relu'))
model.add(Dense(16,activation='sigmoid'))

输出

Metal device set to: Apple M1
2021-09-24 18:18:51.205006: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:305] Could not identify NUMA node of platform GPU ID 0, defaulting to 0. Your kernel may not have been built with NUMA support.
2021-09-24 18:18:51.205728: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:271] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 0 MB memory) -> physical PluggableDevice (device: 0, name: METAL, pci bus id: <undefined>)

这是否意味着我没有使用m1的GPU？

Answer 1

苹果硅是UMA，而不是NUMA。这是一个统一的内存体系结构。所以忽略这条信息。TF将工作，尽管这令人讨厌的信息信息。

Answer 2

您正在GPU上运行，但是如果尝试使用Adam优化器，您可能会发现问题。如果你真的尝试了SGD而不是。

如果要检查培训时是否使用GPU，可以将%GPU和GPU时间添加到活动监视器中。