Suppress tensorflow-metal initialization messages

Question

Created Apr ’23

Replies 1

Boosts 4

Participants 3

Is there a way to stop the following messages when initializing tensorflow with "tensorflow-metal" installed?

Metal device set to: Apple M2 Max
systemMemory: ###.00 GB
maxCacheSize: ###.00 GB

Searching the net, I only came across the following link without any solution: https://stackoverflow.com/questions/73615934/suppress-tensorflow-metal-initialization-messages

I have tried everything but was unable, and in my case these messages are cluttering the console. I believe these messages originate in the following binary file:

&#x2F;Users&#x2F;staff&#x2F;venv-metal&#x2F;lib&#x2F;python3.10&#x2F;site-packages&#x2F;tensorflow-plugins&#x2F;libmetal_plugin.dylib

Thanks to anyone who can share the solution!

Boost

Answer 1

Frameworks Engineer OP

Apple

Apr ’23

Hi @GoodGuyGreg!

Thanks for reporting this. The issue seems to be that the above message gets streamed directly to the standard output so it doesn't follow the same logic as TensorFlow logging messages. I'll create a work item for us to move the printout to be streamed into the TensorFlow logger instead so it can be disabled along the usual TensorfFlow information printouts.

2