How to install tensorflow with GPU support in Macbook Pro M2 chip?

Question

kush0123 OP

Created Feb ’23

Replies 1

Boosts 0

Participants 2

I am having Macbook Pro M2 chip and I want to run Tensorflow with GPU support in it. Can anyone tell me how to do that?

Boost

Answer 1

poltomo OP

Nov ’23

https://developer.apple.com/metal/tensorflow-plugin/

I tried this, but no luck.

2023-11-08 17:40:02.418320: I metal_plugin/src/device/metal_device.cc:1154] Metal device set to: Apple M2 Pro 2023-11-08 17:40:02.418348: I metal_plugin/src/device/metal_device.cc:296] systemMemory: 16.00 GB 2023-11-08 17:40:02.418354: I metal_plugin/src/device/metal_device.cc:313] maxCacheSize: 5.33 GB 2023-11-08 17:40:02.418390: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:306] Could not identify NUMA node of platform GPU ID 0, defaulting to 0. Your kernel may not have been built with NUMA support. 2023-11-08 17:40:02.418411: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:272] 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>)

I think the gpu is not being recognized.

0