MacOS with external GPU for machine learning

Greetings!

I have a MacBook Pro (2018) running macOS Monterey and was wondering if it could be repurposed for machine learning work. I have a Razer Core X, which would enable me to use AMD Radeon 6900 XT as a eGPU supported configuration.

Before doing any shopping, my question is whether the usage of TensorFlow is feasible with this setup. I am aware of tensorflow-metal, but if I understand correctly, only natively supported Mac GPUs can be used with it (not eGPUs)? PlaidML is also an option, but it still not clear which external GPUs are supported.

Any guidance on this topic will be appreciated.

Replies

Any update on this? I’m in the same situation. Are there any other good substitutes for Tensorflow that could work?