Train Tensorflow models using Neural Engine on M2 chip

Question

gn48 OP

Created Apr ’23

Replies 3

Boosts 0

Views 3.6k

Participants 3

Hi!

I've just got a MacBook Air with M2 chip. I'm doing some research on ML and I was wondering if it is possible to use Neural Engine as an accelerator for training. If it is possible, where can I find ressources on how to enable it and how to check it?

Thank you!

Answered by CMDdev in 750964022

There is a Metal TensorFlow Plugin available, which accelerates model training using your Mac's GPU.

Boost

Answer 1

CMDdev OP

Apr ’23

Accepted Answer

There is a Metal TensorFlow Plugin available, which accelerates model training using your Mac's GPU.

1

Answer 2

gn48 OP

Apr ’23

I saw that but I was wondering if it only use GPU or if it gives us access to the Neural Engine

0

Answer 3

NFMi OP

Jul ’23

Hi,

I was wondering the same, did you find an answer/solution to that?

0