Hi all,
I'd installed following the steps from https://developer.apple.com/metal/tensorflow-plugin/
But while I'm running my training I get the log below and the performance is poor than CPU.
Init Plugin
Init Graph Optimizer
Init Kernel
Metal device set to: Apple M1
2021-07-17 18:40:53.716687: 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-07-17 18:40:53.716767: 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>)
My versions:
% python -V
Python 3.9.6
% pip freeze
absl-py @ file:///home/conda/feedstock_root/build_artifacts/absl-py_1606234718434/work
astunparse @ file:///home/conda/feedstock_root/build_artifacts/astunparse_1610696312422/work
cached-property @ file:///home/conda/feedstock_root/build_artifacts/cached_property_1615209429212/work
cachetools==4.2.2
certifi==2021.5.30
charset-normalizer==2.0.3
cloudpickle==1.6.0
flatbuffers==1.12
gast @ file:///home/conda/feedstock_root/build_artifacts/gast_1596839682936/work
google-auth==1.33.0
google-auth-oauthlib==0.4.4
google-pasta==0.2.0
grpcio @ file:///Users/runner/miniforge3/conda-bld/grpcio_1610588577338/work
gym==0.18.3
h5py @ file:///Users/runner/miniforge3/conda-bld/h5py_1609497507927/work
idna==3.2
keras-nightly==2.5.0.dev2021032900
Keras-Preprocessing @ file:///home/conda/feedstock_root/build_artifacts/keras-preprocessing_1610713559828/work
Markdown==3.3.4
numpy @ file:///Users/runner/miniforge3/conda-bld/numpy_1610324554245/work
oauthlib==3.1.1
opt-einsum @ file:///home/conda/feedstock_root/build_artifacts/opt_einsum_1617859230218/work
Pillow==8.2.0
protobuf==3.17.2
pyasn1==0.4.8
pyasn1-modules==0.2.8
pyglet==1.5.15
requests==2.26.0
requests-oauthlib==1.3.0
rsa==4.7.2
scipy @ file:///Users/runner/miniforge3/conda-bld/scipy_1624824941870/work
six @ file:///home/conda/feedstock_root/build_artifacts/six_1590081179328/work
tensorboard==2.5.0
tensorboard-data-server==0.6.1
tensorboard-plugin-wit==1.8.0
tensorflow-estimator==2.5.0
tensorflow-macos==2.5.0
tensorflow-metal==0.1.1
termcolor==1.1.0
typing-extensions @ file:///home/conda/feedstock_root/build_artifacts/typing_extensions_1602702424206/work
urllib3==1.26.6
Werkzeug==2.0.1
wrapt @ file:///Users/runner/miniforge3/conda-bld/wrapt_1624972047019/work
How to speed up using the GPU?
Best Regards,
Fernando