Tensorflow acceleration on macOS

Would it be possible to use GPU acceleration when training a Tensorflow model on macOS? How’s the performance when we training the same model on an Apple-chip platform?
Up vote post of StargazerVia Down vote post of StargazerVia
1.4k views
Posted by
StargazerVia
Reply
Add a Comment

Replies

You can try PlaidML project.

But I'm recommending use Google Colab for training models.
Posted by
alocorn
Up vote reply of alocorn Down vote reply of alocorn
Add a Comment
Tensorflow is now being developed for Mac including the M1.

https://github.com/apple/tensorflow_macos

Posted by
Prof. Nilsson
Up vote reply of Prof. Nilsson Down vote reply of Prof. Nilsson
Add a Comment

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

Posted by
Frameworks Engineer
Up vote reply of Frameworks Engineer Down vote reply of Frameworks Engineer
Add a Comment

Hi @StargazerVia, We have introduced Metal Pluggable Device (thanks @dark_king for providing the link as well) interface for Tensorflow. This provides GPU acceleration on Mac platform for Tensorflow models. Please give it a try and let us know if you run into any issues.

Posted by
Frameworks Engineer
Up vote reply of Frameworks Engineer Down vote reply of Frameworks Engineer
Add a Comment