I have the exact same problem. Could not identify NUMA node of platform GPU ID 0, defaulting to 0. Your kernel may not have been built with NUMA support. Some people actually think it is just a warning. It is NOT a warning, training doesn't even get started, i gets STUCK, even if i leave it for 30 minutes. When i switch to CPU, it works fine. But isn't the whole purpose of high-performance GPU to train model on GPU? Apple support, are you reading this? Tensorflow does NOT work on GPU on M1 !!!

I get NUMA node error message and everything gets stucked! Training doesn't start at all (i waited for 30 minutes and nothing happens). When i switch to CPU, it works properly and there is no NUMA node error message, but of course CPUs are slower compared to GPU in general. It's extremely interesting that noone from Apple or from Tensorflow confirmed this is an issue... and i see this same BUG report all over the internet. It's horrible!! PLEASE FIX THIS, otherwise my new MBP 16 max is useless.

I would be happy to ignore this error message, the problem is that model.fit gets totally stucked, it doesn't even start traning. NADA. If i switch to CPU, then everything works fine. Switch back to GPU, it prints out that error message and then it gets stuck on model.fit. The whole internet is flooded with this exact same problem and neither Apple or Tensoflow takes it seriously. If somone would properly classify this as a bug, then we would know it will get solved. Back to PC i guess!?