Apple SiliconのMacでGPUをTensorflowで用いる方法

注意
こちらの記事は公式の情報源に依っているものではなく、確認が必要な内容を含みます。
詳しい方の指摘・補足をお待ちしております。

結論

Apple自身が作っているtensorflow-metalを使う。

pip install tensorflow-macos
pip install tensorflow-metal

今はもうやらなくて良さそうなこと

Conda経由でのtensorflow-depsのインストール
https://anaconda.org/apple/tensorflow-deps
Anaconda上は、2年前から更新がなされていない

conda install apple tensorflow-deps

もう少し調べたいこと

Apple SiliconのMac上のDockerで、Ubuntu環境を立ち上げた際にもGPUを動かせるか。
→試したらインストールできないとエラーが出た。tensorflow-metalがOS依存しているか、OS判定に引っかかったか。